数据共有三类,及对应的预测值。分类为1,2,3,想要比较各个分类的ROC曲线,最终实现的代码如下。
library("pROC")
all[which(all$class!=3),]->all_1v2 #all$class 列为真实分类标签
roc1 <- roc(all_1v2$class,all_1v2$predict) #all$predict 列为预测值
plot(roc1, col="blue")
all[which(all$class!=2),]->all_1v3
roc2 <- roc(all_1v3$class,all_1v3$predict)
plot.roc(roc2, add=TRUE, col="red")
all[which(all$class!=1),]->all_2v3
roc3 <- roc(all_2v3$class,all_2v3$predict)
plot.roc(roc3, add=TRUE, col="black")
网友评论