c RMSE на, например Tweedie или MAPE, то меняется метрика по которой оценивается моя модель? Не смог разобраться в доке по eval_metric параметру (если это то, что мне нужно, вообще)
Код простой
boost = CatBoostRegressor(**params) # здесь задаю loss_function
train = Pool(data=X_train, label=y_train, cat_features=cat_cols)
test = Pool(data=X_test, label=y_test, cat_features=cat_cols)
boost.fit(
train, eval_set=test, use_best_model=True, early_stopping_rounds=early_stop
)
pred = boost.predict(X_test)
Да, самое главное, что меняется метрика, которая считается для теста. Это влияет на early stopping rounds, а это влияет на количество деревьев
Обсуждают сегодня