使用multipleROC快速绘制ROC曲线

之前我写过一篇使用pROC包画好看的ROC曲线的教程,那篇教程中使用的是pROC,这个包可以快速拟合ROC曲线,然而这个包需要提前进行运算结果,并且不能直接显示AUC值等,今天推荐一个另一个绘制ROC的包multipleROC,顾名思义,这个包是可以一次性绘制多条ROC曲线的,并且也是基于ggplot2。

目前这个包作者没有上传CRAN或BiocManager,只能通过Github安装,地址为https://github.com/cardiomoon/multipleROC

安装multipleROC

remotes::install_github("cardiomoon/multipleROC")

如果无法访问GitHub,也可以导入到Gitee后进行安装

remotes::install_git("https://gitee.com/swcyo/multipleROC/")

数据演示

我们使用仙桃学术上的一个诊断性ROC示例数据为例进行演示(下载请点击xlxs链接)。

library(readxl) 
ROC <- read_excel("~/Desktop/ROC曲线.xlsx")

探索性分析

我们可以事先看一下group1和group2两组在a变量中的差别,使用webr包,先看看结果如何

library(webr)
library(ggplot2)
library(dplyr)
library(tidyr) 
ROC %>% 
  group_by(outcome) %>% 
  numSummaryTable(a)
outcome n mean sd median trimmed mad min max range
group1 40.00 1.51 0.55 1.45 1.51 0.63 0.59 2.44 1.86
group2 32.00 1.00 0.55 1.00 0.99 0.63 0.13 1.99 1.86

也可以使用箱示图和密度图进行展示,见Figure 1所示。

p1<- ggplot(data=ROC)+geom_density(aes(x=a,fill=outcome),alpha=0.5)  
p2<-ggplot(data=ROC)+geom_boxplot(aes(x=outcome,y=a,fill=outcome),alpha=0.5)  
cowplot::plot_grid(p1,p2)
Figure 1: group1和group2两组在a变量中的差别

同法可以显示b和c变量的结果,我们暂时以boxplot展示

p3<-ggplot(data=ROC)+geom_boxplot(aes(x=outcome,y=b,fill=outcome),alpha=0.5) 
p4<-ggplot(data=ROC)+geom_boxplot(aes(x=outcome,y=c,fill=outcome),alpha=0.5) 
cowplot::plot_grid(p2,p3,p4,labels = "AUTO",nrow = 1)
Figure 2: group1和group2两组在三变量中的差别

-- 虽然探索性分析可以判断两组的差异,但是无法确定最佳截断值,也无妨评估预测效能。

ROC曲线的绘制

绘制ROC曲线是确定最佳截断值的有用方法之一。您可以使用以下R命令执行ROC分析。下面的R命令使一个类为multipleROC的对象,并进行绘图。
由于默认的函数中分组需为0和1,因此需要将group1和group2进行赋值,我们将group1定义为0,group2定义为1,我们绘制a变量在两组中的ROC图片,我们可以使用multipleROC()语句一步计算,可以看到最佳截断值,AUC值,另外敏感度、特异度都是可以直接显示的,见Figure 3所示。。

ROC$group<-ifelse(ROC$outcome=='group1',0,1) # 将group1定义为0,否则为1 
library(multipleROC) 
a=multipleROC(group~a,data=ROC)
Figure 3: a变量在两组的ROC曲线

如果不想显示那么多结果的话,也可以plot_ROC()函数一个个设置是否显示

plot_ROC(a, 
         show.points = TRUE, 
         show.eta = TRUE, 
         show.sens = TRUE, 
         show.AUC = TRUE, 
         facet = FALSE )

AUC和p值

在Figure 3的右下角,您可以看到曲线下面积(AUC)和Wilcoxon秩和检验的p值。p值来自以下计算结果。

wilcox.test(ROC$a,ROC$group)

##  
##  Wilcoxon rank sum test with continuity correction 
##  
## data:  ROC$a and ROC$group
## W = 4416, p-value = 1.294e-13
## alternative hypothesis: true location shift is not equal to 0

AUC值则通过multipleROC包的simpleAUC()函数进行运算,函数如下:

simpleAUC <- function(df){
     df=df[order(df$x,decreasing=TRUE),]
     TPR=df$sens
     FPR=df$fpr

     dFPR <- c(diff(FPR), 0)
     dTPR <- c(diff(TPR), 0)

     sum(TPR * dFPR) + sum(dTPR * dFPR)/2
}

那么,我们直接直接只有simpleAUC(adf) 进行提取,或者简单的的aauc直接看到完整的AUC值

simpleAUC(a$df)  ## 函数法
## [1] 0.7328125

a$auc # 直接提取法
## [1] 0.7328125

同样的,我们直接提取截断点(cutpoint)和最佳截断值(Optimal Cutoff value)

a$cutpoint
## [1] 0.5136663

a$cutoff
##  a ## 54 1.082828

将结果转换为pROC对象

如果你更习惯pROC的结果,使用multipleROC2roc()函数,可以直接将结果转换为 pROC的roc 对象

a2<-multipleROC2roc(a)
## Setting levels: control = 0, case = 1
## Setting direction: controls < cases

class(a) ##a的类型为multipleROC
## [1] "multipleROC"

class(a2) ##a2已经转换为roc的类型了
## [1] "roc"

pROC::auc(a2) ## 我们用pROC看auc的结果
## Area under the curve: 0.7328

我们可以使用pROC的绘图函数对a2进行绘图了,我们比较以下两种结果吧,见Figure 4所示。

library(pROC)
p5<-ggroc(a2, legacy.axes = TRUE)+
  geom_segment(aes(x = 0, xend = 1, y = 0, yend = 1), 
               color="darkgrey",linetype=4)+
  theme_bw()+ggtitle("pROC")
p6<-plot(a)+ggtitle("multipleROC")
cowplot::plot_grid(p5,p6)
Figure 4: pROC和multipleROC的结果对比

多个ROC曲线的绘制

可以用多个函数进行多个ROC的曲线,可以使用plot_ROC(list())一个个绘制曲线,见Figure 5所示。

a=multipleROC(group~a,data=ROC,plot=FALSE)
b=multipleROC(group~b,data=ROC,plot=FALSE)
c=multipleROC(group~c,data=ROC,plot=FALSE)
plot_ROC(list(a,b,c),
         show.eta=FALSE,#不显示截点
         show.sens=FALSE #不显示各种率
         )
Figure 5: 三条曲线同时绘制

当然,如果你不想写那么多代码的话,也可以直接使用plot_ROC2()函数直接绘制,是不是简单的多。

plot_ROC2(yvar="group",xvars=c("a","b","c"),dataname="ROC")
Figure 6: 三条曲线同时绘制简单函数

分面显示

将三张图放在一起,可以看到数值重叠影响颜值,我们可以用ggplot2的facet进行分面绘制。

plot_ROC(list(a,b,c),facet=TRUE)
Figure 7: 三条曲线的分面显示

可以发现分面的标签默认是1,2,3,我们可以使用Y叔团队开发的ggfun这个包的facet_set()函数进行快速的修改

library(ggfun) ## 必须要先安装好
plot_ROC(list(a,b,c),facet=TRUE)+
  facet_set(label=c(`1`="a", `2`="b", `3`="c"))
分面显示该标签名称

换一种分面显示

使用ggplot2包的facet_grid可以换一个分面显示方式

plot_ROC(list(a,b,c))+facet_grid(no~.)+
  facet_set(label=c(`1`="a", `2`="b", `3`="c"))
换一个分面显示方式

由于是基于ggplot2语句,所以我们可以使用ggtitle添加标题,还可以更换主题等等。。。

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 217,826评论 6 506
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 92,968评论 3 395
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 164,234评论 0 354
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 58,562评论 1 293
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 67,611评论 6 392
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 51,482评论 1 302
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 40,271评论 3 418
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 39,166评论 0 276
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 45,608评论 1 314
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 37,814评论 3 336
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 39,926评论 1 348
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 35,644评论 5 346
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 41,249评论 3 329
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 31,866评论 0 22
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 32,991评论 1 269
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 48,063评论 3 370
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 44,871评论 2 354