R h2o: как реализовать пользовательский параметр stopping_metric для GBM - PullRequest
Новая
       32

R h2o: как реализовать пользовательский параметр stopping_metric для GBM

0 голосов
/ 30 января 2019

Я обнаружил, что теперь можно использовать stopping_metric = custom в h2o v3.22.1.1 (недоступно в v3.10.0.9), однако я нигде не нашел, как реализовать это в R.

это игрушечная версия проблемы. 

library(h2o)
h2o.init()
x <- data.frame(
   x = rnorm(1000),
   z = rnorm(1000), 
   y = factor(sample(0:1, 1000, replace = T))
)
train <- as.h2o(x)
h2o.gbm(x = c('x','z'), y = 'y', training_frame = train, stopping_metric = 'custom', stopping_rounds = 3)

я получаю следующую ошибку: 

Error in .h2o.doSafeREST(h2oRestApiVersion = h2oRestApiVersion, urlSuffix = page,  : 


ERROR MESSAGE:

Illegal argument(s) for GBM model: GBM_model_R_1548836369139_123.  Details: ERRR on field: _stopping_metric: Custom metric function needs to be defined in order to use it for early stopping.

как я могу определить custom stopping_metric для GBM

1 Ответ

0 голосов
/ 31 января 2019

В настоящее время недоступно в R API.Если вас интересует эта функциональность, я бы создал JIRA с вашим вариантом использования здесь .Вы можете использовать другой SO вопрос, на который указал Даррен , для получения информации о том, как использовать эту функцию через Python API.

...