Сообщение об ошибке при построении ROC объекта модели H2O в R

Question

При попытке построить кривую ROC для объекта модели H2O в R, я продолжаю получать следующее сообщение об ошибке:

"Ошибка в as.double (y): невозможно привести тип 'S4' к векторутип 'double' "

Мой код выглядит следующим образом:

drf1 <- h2o.randomForest(x=x,y=y,training_frame = train,validation_frame = valid, nfolds = nfolds, fold_assignment = "Modulo",keep_cross_validation_predictions = TRUE,seed = 1)

plot((h2o.performance(drf1,valid = T)), type = "roc")

Я следовал рекомендациям, найденным здесь: Как напрямую построить ROC объекта модели H2O в R

Любая помощь будет принята с благодарностью!

Answer 1

Из-за ошибки, я думаю, ваша переменная ответа не является двоичной.Вы можете изменить свою переменную ответа на фактор, прежде чем поместить ее в модель.т. е.

df$y <- as.factor(df$y)

"ROC - это графический график, который иллюстрирует диагностические возможности системы двоичного классификатора , поскольку ее порог различения варьируется".

источник: ROC wiki