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