max_depth

Оценка качества, расчет accuracy_score при разной максимальной глубине

df_model_train, df_validation = train_test_split(df, test_size=0.25, random_state=5678)

features_model_train = df_model_train.drop(['target_col'], axis=1)
target_model_train = df_model_train['target_col']
features_validation = df_validation.drop(['target_col'], axis=1)
target_validation = df_validation['target_col']

Максимальная глубина

max_depth

Настройка глубины дерева max_depth

model = DecisionTreeClassifier(random_state=5678, max_depth=4)

Подписка на RSS - max_depth