evalml.pipelines.components.RandomForestClassifier

Inheritance diagram of RandomForestClassifier
class evalml.pipelines.components.RandomForestClassifier(n_estimators=100, max_depth=6, n_jobs=-1, random_state=0, **kwargs)[source]

Random Forest Classifier.

name = 'Random Forest Classifier'
model_family = 'random_forest'
supported_problem_types = [<ProblemTypes.BINARY: 'binary'>, <ProblemTypes.MULTICLASS: 'multiclass'>]
hyperparameter_ranges = {'max_depth': Integer(low=1, high=10, prior='uniform', transform='identity'), 'n_estimators': Integer(low=10, high=1000, prior='uniform', transform='identity')}
default_parameters = {'max_depth': 6, 'n_estimators': 100, 'n_jobs': -1}

Instance attributes

feature_importance

Returns importance associated with each feature.

parameters

Returns the parameters which were used to initialize the component

Methods:

__init__

Initialize self.

clone

Constructs a new component with the same parameters

describe

Describe a component and its parameters

fit

Fits component to data

predict

Make predictions using selected features.

predict_proba

Make probability estimates for labels.