converters.sklearn._gradient_boosting_regressor

Module Contents

Functions

convert(model,input_features,output_features) Convert a boosted tree model to protobuf format.
convert(model, input_features, output_features)

Convert a boosted tree model to protobuf format.

decision_tree : GradientBoostingRegressor
A trained scikit-learn tree model.
input_feature: [str]
Name of the input columns.
output_features: str
Name of the output column.
model_spec: An object of type Model_pb.
Protobuf representation of the model