R语言绘图：ggplot2绘制ROC

最新推荐文章于 2024-07-09 08:43:29 发布

aiairen0123

最新推荐文章于 2024-07-09 08:43:29 发布

阅读量2.7k

点赞数

文章标签： r语言

原文链接：http://www.cnblogs.com/xihehe/p/8298029.html

版权

本文介绍如何使用R语言的ggplot2包绘制ROC曲线，并计算AUC值，通过一个函数实现预测值与真实值之间的ROC曲线绘制，包括线条、填充色及标题等元素的定制。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

使用ggplot2包绘制ROC曲线

rocplot<- function(pred, truth, ...){
    predob<- prediction(pred, truth)
    #打印AUc
    perf.auc<- performance(predob, measure = 'auc', x.measure = 'cutoff')
    #
    perf<- performance(predob, 'tpr','fpr')
    df<- data.frame(x = attributes(perf)$x.values[[1]],y = attributes(perf)$y.values[[1]])  
    p    <- ggplot(data = df)
    p + geom_line(aes(x,y),colour = "yellowgreen",size = 1) + 
        geom_ribbon(aes(x,ymin = 0,ymax = y),fill = alpha("yellowgreen",0.5)) +
        labs(title = paste("ROC Curve & AUC:",(perf.auc@y.values))) + 
        xlab("Specificity") +
        ylab("Sensitivity") +
        theme(plot.title = element_text(size = 17)) 
}

rocplot((model1.prob), data2[test, ]$results)

转载于:https://www.cnblogs.com/xihehe/p/8298029.html