изменить гиперпараметры RandomForestRegressor по умолчанию

lab_2/master
syropiatovvv 2 дней назад
Родитель 8c3e3c1588
Сommit ce02a6966b
Подписано: syropiatovvv
Идентификатор GPG ключа: 297380B8143A31BD

@ -242,7 +242,9 @@ def build_categorical_features_encoder_target(*, random_state=None):
# %%
def build_regressor(*, random_state=None):
return sklearn.ensemble.RandomForestRegressor(
10, criterion='squared_error', max_features='sqrt', random_state=random_state,
10, criterion='squared_error',
max_depth=8, max_features='sqrt',
random_state=random_state,
)

Загрузка…
Отмена
Сохранить