evalml.pipelines.components.DecisionTreeClassifier

class evalml.pipelines.components.DecisionTreeClassifier(criterion='gini', max_features='auto', max_depth=6, min_samples_split=2, min_weight_fraction_leaf=0.0, random_state=0, **kwargs)[source]

Decision Tree Classifier.

name = 'Decision Tree Classifier'
model_family = 'decision_tree'
supported_problem_types = [<ProblemTypes.BINARY: 'binary'>, <ProblemTypes.MULTICLASS: 'multiclass'>, <ProblemTypes.TIME_SERIES_BINARY: 'time series binary'>, <ProblemTypes.TIME_SERIES_MULTICLASS: 'time series multiclass'>]
hyperparameter_ranges = {'criterion': ['gini', 'entropy'], 'max_depth': Integer(low=4, high=10, prior='uniform', transform='identity'), 'max_features': ['auto', 'sqrt', 'log2']}
default_parameters = {'criterion': 'gini', 'max_depth': 6, 'max_features': 'auto', 'min_samples_split': 2, 'min_weight_fraction_leaf': 0.0}

Instance attributes

feature_importance

Returns importance associated with each feature.

needs_fitting

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

load

Loads component at file path

predict

Make predictions using selected features.

predict_proba

Make probability estimates for labels.

save

Saves component at file path

Class Inheritance

Inheritance diagram of DecisionTreeClassifier