converters.sklearn._random_forest_regressor

Module Contents

Functions

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

Convert a boosted tree model to protobuf format.

decision_tree : RandomForestRegressor
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