converters.sklearn._gradient_boosting_classifier

Module Contents

Functions

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

Convert a boosted tree model to protobuf format.

decision_tree : GradientBoostingClassifier
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
supports_output_scores(model)
get_output_classes(model)