streamline.models.decision_tree module

class streamline.models.decision_tree.DecisionTreeClassifier(cv_folds=3, scoring_metric='balanced_accuracy', metric_direction='maximize', random_state=None, cv=None, n_jobs=None)

Bases: BaseModel, ABC

Base Model Class for all ML Models

Parameters:

• model –

• model_name –

• cv_folds –

• scoring_metric –

• metric_direction –

• random_state –

• cv –

• sampler –

• n_jobs –

color = 'yellow'

model_name = 'Decision Tree'

objective(trial, params=None)

Unimplemented objective function stub, needs to be overridden :param trial: optuna trial object :param params: dict of optional params or None

small_name = 'DT'