concrete.ml.sklearn.tree
module concrete.ml.sklearn.tree
Implement the sklearn tree models.
class DecisionTreeClassifier
Implements the sklearn DecisionTreeClassifier.
method __init__
__init__(
criterion='gini',
splitter='best',
max_depth=None,
min_samples_split=2,
min_samples_leaf=1,
min_weight_fraction_leaf=0.0,
max_features=None,
random_state=None,
max_leaf_nodes=None,
min_impurity_decrease=0.0,
class_weight=None,
ccp_alpha: float = 0.0,
n_bits: int = 6
)
Initialize the DecisionTreeClassifier.
noqa: DAR101
property onnx_model
Get the ONNX model.
.. # noqa: DAR201
Returns:
onnx.ModelProto: the ONNX model
class DecisionTreeRegressor
Implements the sklearn DecisionTreeClassifier.
method __init__
__init__(
criterion='squared_error',
splitter='best',
max_depth=None,
min_samples_split=2,
min_samples_leaf=1,
min_weight_fraction_leaf=0.0,
max_features=None,
random_state=None,
max_leaf_nodes=None,
min_impurity_decrease=0.0,
ccp_alpha=0.0,
n_bits: int = 6
)
Initialize the DecisionTreeRegressor.
noqa: DAR101
property onnx_model
Get the ONNX model.
.. # noqa: DAR201
Returns:
onnx.ModelProto: the ONNX model