turicreate.decision_tree_classifier.DecisionTreeClassifier

class turicreate.decision_tree_classifier.DecisionTreeClassifier(proxy)

Special case of gradient boosted trees with the number of trees set to 1.

The decision tree model can be used as a classifier for predictive tasks. Different from linear models like logistic regression or SVM, this algorithm can model non-linear interactions between the features and the target. This model is suitable for handling numerical features and categorical features with tens of categories but is less suitable for highly sparse features (text data), or with categorical variables that encode a large number of categories.

This model cannot be constructed directly. Instead, use turicreate.decision_tree_classifier.create() to create an instance of this model. Additional details on parameter options and code samples are available in the documentation for the create function.

See also

create

Methods

DecisionTreeClassifier.classify(self, dataset) Return a classification, for each example in the dataset, using the trained model.
DecisionTreeClassifier.evaluate(self, dataset) Evaluate the model by making predictions of target values and comparing these to actual values.
DecisionTreeClassifier.export_coreml(self, …) Export the model in Core ML format.
DecisionTreeClassifier.extract_features(…) For each example in the dataset, extract the leaf indices of each tree as features.
DecisionTreeClassifier.get_feature_importance(self) Get the importance of features used by the model.
DecisionTreeClassifier.predict(self, dataset) A flexible and advanced prediction API.
DecisionTreeClassifier.predict_topk(self, …) Return top-k predictions for the dataset, using the trained model.
DecisionTreeClassifier.save(self, location) Save the model.
DecisionTreeClassifier.summary(self[, output]) Print a summary of the model.