当前位置:  开发笔记 > 编程语言 > 正文

plot.roc用于pROC包中的multiclass.roc?

如何解决《plot.roc用于pROC包中的multiclass.roc?》经验,为你挑选了1个好方法。

我试图绘制多类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曲线?



1> 小智..:

检查名称roc.multi,你应该找到一个名字rocs,它存储每个类的单独roc曲线信息.

所以你可以使用plot.roclines.roc可视化所有这些:

rs <- roc.multi[['rocs']]
plot.roc(rs[[1]])
sapply(2:length(rs),function(i) lines.roc(rs[[i]],col=i))

推荐阅读
乐韵答题
这个屌丝很懒,什么也没留下!
DevBox开发工具箱 | 专业的在线开发工具网站    京公网安备 11010802040832号  |  京ICP备19059560号-6
Copyright © 1998 - 2020 DevBox.CN. All Rights Reserved devBox.cn 开发工具箱 版权所有