我试图绘制多类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))