Как получить индекс Джини после выполнения поиска по сетке в настройке гиперпараметров GBM для функции потери твиди?
Я занимаюсь настройкой гиперпараметра для модели gbm в H2o, и поскольку моя функция потерь - Tweedie, я не хочу рассматривать mse как критерий выбора модели.
В документации H2o говорится, что индекс Джини может быть рассчитан как для моделей регрессии, так и для моделей классификации, однако, когда я пытаюсь получить его для моей модели регрессии Твиди, он возвращает ноль. Ниже показано, как я могу получить лучшую модель и оценить ее на тестовом наборе.
gbm_sorted_grid <- h2o.getGrid(grid_id = "grid_hp4", sort_by =
"residual_deviance")
best_model <- h2o.getModel(gbm_sorted_grid@model_ids[[1]])
perf <- h2o.performance(best_model, newdata = lrs_test)
h2o.giniCoef(perf)
Null
И когда я пытаюсь код ниже, я получаю ошибку ниже:
h2o.giniCoef(best_model) : No Gini for H2ORegressionModel
Это работает только на дистрибутив Бернулли?
1 ответ
GiniCoef в H2O-3 поддерживает только проблемы двоичной классификации, это ошибка в документации. Я создал билет jira, чтобы руководство пользователя обновлялось. Спасибо за освещение проблемы!