converters.sklearn._decision_tree_regressor

Module Contents

Functions

convert(model,feature_names,target) Convert a decision tree model to protobuf format.
get_input_dimension(model)
convert(model, feature_names, target)

Convert a decision tree model to protobuf format.

decision_tree : DecisionTreeRegressor
A trained scikit-learn tree model.
feature_names: [str]
Name of the input columns.
target: str
Name of the output column.
model_spec: An object of type Model_pb.
Protobuf representation of the model
get_input_dimension(model)