converters.sklearn._decision_tree_classifier

Module Contents

Functions

convert(model,input_name,output_features) Convert a decision tree model to protobuf format.
supports_output_scores(model)
get_output_classes(model)
get_input_dimension(model)
convert(model, input_name, output_features)

Convert a decision tree model to protobuf format.

decision_tree : DecisionTreeClassifier
A trained scikit-learn tree model.
input_name: str
Name of the input columns.
output_name: str
Name of the output columns.
model_spec: An object of type Model_pb.
Protobuf representation of the model
supports_output_scores(model)
get_output_classes(model)
get_input_dimension(model)