在R

时间:2018-11-28 04:07:47

标签: r h2o roc

试图在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

任何帮助将不胜感激!

1 个答案:

答案 0 :(得分:1)

从错误中,我认为您的响应变量不是二进制的。您可以在将其放入模型之前更改响应变量以进行分解。即

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

“ ROC是一个图形图,它说明了二进制分类器系统的鉴别阈值变化时的诊断能力。”

来源: ROC wiki