试图在R中绘制H2O模型对象的ROC曲线,但是,我仍然收到以下错误消息:
“ 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")
我遵循了在这里找到的建议:How to directly plot ROC of h2o model object in R
任何帮助将不胜感激!
答案 0 :(得分:1)
从错误中,我认为您的响应变量不是二进制的。您可以在将其放入模型之前更改响应变量以进行分解。即
df$y <- as.factor(df$y)
“ ROC是一个图形图,它说明了二进制分类器系统的鉴别阈值变化时的诊断能力。”
来源: ROC wiki