我正在尝试绘制多类 ROC 曲线,但我在 pROC 包中没有发现任何富有成效的东西。这是一些启动代码:
data(iris)
library(randomForest)
library(pROC)
set.seed(1000)
# 3-class in response variable
rf = randomForest(Species~., data = iris, ntree = 100)
# predict(.., type = 'prob') returns a probability matrix
predictions <- as.numeric(predict(rf, iris, type = 'response'))
roc.multi <- multiclass.roc(iris$Species, predictions)
auc(roc.multi)
如何绘制各个类别的 ROC 曲线?
最佳答案
检查 roc.multi
的名称,您应该找到一个名为 rocs
的名称,它存储每个类的单独 roc 曲线信息。
所以你可以使用 plot.roc
和 lines.roc
来可视化它们:
rs <- roc.multi[['rocs']]
plot.roc(rs[[1]])
sapply(2:length(rs),function(i) lines.roc(rs[[i]],col=i))
关于r - pROC 包中 multiclass.roc 的 plot.roc ?,我们在Stack Overflow上找到一个类似的问题:https://stackoverflow.com/questions/34169774/