turicreate.boosted_trees_regression.BoostedTreesRegression

class turicreate.boosted_trees_regression.BoostedTreesRegression(proxy)

Encapsulates gradient boosted trees for regression tasks.

The prediction is based on a collection of base learners, regression trees.

Different from linear models, e.g. linear regression, the gradient boost trees model is able to model non-linear interactions between the features and the target using decision trees as the subroutine. It is good for handling numerical features and categorical features with tens of categories but is less suitable for highly sparse features such as text data.

This model cannot be constructed directly. Instead, use turicreate.boosted_trees_regression.create() to create an instance of this model. A detailed list of parameter options and code samples are available in the documentation for the create function.

See also

create

Methods

BoostedTreesRegression.evaluate(self, dataset) Evaluate the model on the given dataset.
BoostedTreesRegression.export_coreml(self, …) Export the model in Core ML format.
BoostedTreesRegression.extract_features(…) For each example in the dataset, extract the leaf indices of each tree as features.
BoostedTreesRegression.get_feature_importance(self) Get the importance of features used by the model.
BoostedTreesRegression.predict(self, dataset) Predict the target column of the given dataset.
BoostedTreesRegression.save(self, location) Save the model.
BoostedTreesRegression.summary(self[, output]) Print a summary of the model.