turicreate.random_forest_regression.RandomForestRegression

class turicreate.random_forest_regression.RandomForestRegression(proxy)

Encapsulates random forest models for regression tasks.

The prediction is based on a collection of base learners, regression trees and combines them through a technique called random forest.

Different from linear models, e.g. linear regression, the random forests are 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.random_forest_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

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