多分类ROC曲线

2020-07-21  本文已影响0人  阿酒88

数据共有三类,及对应的预测值。分类为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")

上一篇下一篇

猜你喜欢

热点阅读