用 pROC R 绘制 ROC 曲线

Question

我使用 randomForest 构建了一个文本分类器，因此为了评估它，我尝试使用 pROC 包创建 ROC 曲线。

这里是代码：

ndsi.forest <- randomForest(tf.idf[train.index, ], as.factor(train$Note.Reco[train.index]), ntree = 100)

#predict with test data
ndsi.pred <-predict(ndsi.forest, newdata = tf.idf[test.index, ], response  = 'class')
pred <- data.frame(ndsi.pred)
result <- data.frame(id = Data_clean$id[test.index], sentiment = pred[ , ])

##"ROC curve"
multiclass.roc(result$sentiment, test$Note.Reco)

我想知道这是否是一种创建情节的方法？使用 pROC 包的 ROC 图？

我尝试使用此代码：

roc(test$Note.Reco, result$sentiment, levels = c(1,2,3,4,5,6,7,8,9,10))

但我收到此错误：

Error in roc.default(test$Note.Reco, result$sentiment, levels = c(1, 2,  : 
  'levels' argument must have length 2

提前谢谢你

Answer 1

据我了解，您有一个多类响应变量（对应于 10 个不同的组）。

ROC - 曲线是为两组的分类定义的，所以 multiclass 的作用是计算“一个组与其他组”的分类。 multiclass.roc 函数不允许您表示曲线，但了解它的作用后，您可以：

1) 考虑与您拥有的组一样多的 roc 曲线。即ROC-曲线分类：

• 第 1 组与非第 1 组
• 第 2 组与非第 2 组
• 。 . .
• 第 10 组与非第 10 组

您可以使用roc 函数来做到这一点。您唯一需要的是将响应向量重新定义为属于第 i 组的个人为 1，为其余个人重新定义为 0。用不同的名称保存每个 roc 对象。

2) 要表示所有曲线，只需对每条曲线使用plot 函数，将plot(..., add=T) 添加到除第一条之外的所有曲线。