concrete.ml.sklearn.rf

module concrete.ml.sklearn.rf

Implements RandomForest models.

---

class RandomForestClassifier

Implements the RandomForest classifier.

method __init__

__init__(
    n_bits: int = 6,
    n_estimators=20,
    criterion='gini',
    max_depth=4,
    min_samples_split=2,
    min_samples_leaf=1,
    min_weight_fraction_leaf=0.0,
    max_features='sqrt',
    max_leaf_nodes=None,
    min_impurity_decrease=0.0,
    bootstrap=True,
    oob_score=False,
    n_jobs=None,
    random_state=None,
    verbose=0,
    warm_start=False,
    class_weight=None,
    ccp_alpha=0.0,
    max_samples=None
)

Initialize the RandomForestClassifier.

noqa: DAR101

---

property onnx_model

Get the ONNX model.

.. # noqa: DAR201

Returns:

• onnx.ModelProto: the ONNX model

---

class RandomForestRegressor

Implements the RandomForest regressor.

method __init__

__init__(
    n_bits: int = 6,
    n_estimators=20,
    criterion='squared_error',
    max_depth=4,
    min_samples_split=2,
    min_samples_leaf=1,
    min_weight_fraction_leaf=0.0,
    max_features='sqrt',
    max_leaf_nodes=None,
    min_impurity_decrease=0.0,
    bootstrap=True,
    oob_score=False,
    n_jobs=None,
    random_state=None,
    verbose=0,
    warm_start=False,
    ccp_alpha=0.0,
    max_samples=None
)

Initialize the RandomForestRegressor.

noqa: DAR101

---

property onnx_model

Get the ONNX model.

.. # noqa: DAR201

Returns:

• onnx.ModelProto: the ONNX model