美文网首页
多分类ROC曲线

多分类ROC曲线

作者: 阿酒88 | 来源:发表于2020-07-21 09:03 被阅读0次

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

    相关文章

      网友评论

          本文标题:多分类ROC曲线

          本文链接:https://www.haomeiwen.com/subject/ibvbkktx.html