计算R中的AUC?
data-mining
machine-learning
r
5
0

给定分数矢量和实际类别标签矢量,如何为R语言或简单英语中的二进制分类器计算单数AUC度量?

“ AUC:更好的措施...”的第9页似乎需要了解类标签,这是MATLAB中的一个我不理解的示例

R(Actual == 1))

因为R(不要与R语言混淆)被定义为向量而是用作函数?

参考资料:
Stack Overflow
收藏
评论
共 10 个回答
高赞 时间 活跃

我通常使用DiagnosisMed软件包中的ROC函数。我喜欢它产生的图。 AUC连同其置信区间一起返回,并且在图中也提到了AUC。

ROC(classLabels,scores,Full=TRUE)
收藏
评论

我发现这里的某些解决方案速度慢和/或令人困惑(其中一些解决方案无法正确处理关系),因此我在R包mltools中编写了自己的基于data.table的函数auc_roc()

library(data.table)
library(mltools)

preds <- c(.1, .3, .3, .9)
actuals <- c(0, 0, 1, 1)

auc_roc(preds, actuals)  # 0.875

auc_roc(preds, actuals, returnDT=TRUE)
   Pred CountFalse CountTrue CumulativeFPR CumulativeTPR AdditionalArea CumulativeArea
1:  0.9          0         1           0.0           0.5          0.000          0.000
2:  0.3          1         1           0.5           1.0          0.375          0.375
3:  0.1          1         0           1.0           1.0          0.500          0.875
收藏
评论

正如其他人所提到的,您可以使用ROCR包来计算AUC。使用ROCR软件包,您还可以绘制ROC曲线,升力曲线和其他模型选择度量。

您可以通过以下方式直接计算AUC,而无需使用任何程序包:使用AUC等于将真实的阳性得分大于真实的阴性得分的概率。

例如,如果pos.scores是包含分数阳性示例的向量,而neg.scores是包含阴性示例的向量,则AUC近似为:

> mean(sample(pos.scores,1000,replace=T) > sample(neg.scores,1000,replace=T))
[1] 0.7261

将给出AUC的近似值。您还可以通过引导来估计AUC的方差:

> aucs = replicate(1000,mean(sample(pos.scores,1000,replace=T) > sample(neg.scores,1000,replace=T)))
收藏
评论

ROCR软件包将在其他统计数据中计算AUC:

auc.tmp <- performance(pred,"auc"); auc <- as.numeric(auc.tmp@y.values)
收藏
评论

您可以在Miron Kursa的此博客文章中了解有关AUROC的更多信息:

https://mbq.me/blog/augh-roc/

他为AUROC提供了快速功能:

# By Miron Kursa https://mbq.me
auroc <- function(score, bool) {
  n1 <- sum(!bool)
  n2 <- sum(bool)
  U  <- sum(rank(score)[!bool]) - n1 * (n1 + 1) / 2
  return(1 - U / n1 / n2)
}

让我们测试一下:

set.seed(42)
score <- rnorm(1e3)
bool  <- sample(c(TRUE, FALSE), 1e3, replace = TRUE)

pROC::auc(bool, score)
mltools::auc_roc(score, bool)
ROCR::performance(ROCR::prediction(score, bool), "auc")@y.values[[1]]
auroc(score, bool)

0.51371668847094
0.51371668847094
0.51371668847094
0.51371668847094

auroc()pROC::auc()computeAUC()快100倍。

auroc()mltools::auc_roc()ROCR::performance()快10倍。

print(microbenchmark(
  pROC::auc(bool, score),
  computeAUC(score[bool], score[!bool]),
  mltools::auc_roc(score, bool),
  ROCR::performance(ROCR::prediction(score, bool), "auc")@y.values,
  auroc(score, bool)
))

Unit: microseconds
                                                             expr       min
                                           pROC::auc(bool, score) 21000.146
                            computeAUC(score[bool], score[!bool]) 11878.605
                                    mltools::auc_roc(score, bool)  5750.651
 ROCR::performance(ROCR::prediction(score, bool), "auc")@y.values  2899.573
                                               auroc(score, bool)   236.531
         lq       mean     median        uq        max neval  cld
 22005.3350 23738.3447 22206.5730 22710.853  32628.347   100    d
 12323.0305 16173.0645 12378.5540 12624.981 233701.511   100   c 
  6186.0245  6495.5158  6325.3955  6573.993  14698.244   100  b  
  3019.6310  3300.1961  3068.0240  3237.534  11995.667   100 ab  
   245.4755   253.1109   251.8505   257.578    300.506   100 a   
收藏
评论

目前,票数最高的答案是不正确的,因为它无视联系。当正负分数相等时,则AUC应为0.5。下面是更正的示例。

computeAUC <- function(pos.scores, neg.scores, n_sample=100000) {
  # Args:
  #   pos.scores: scores of positive observations
  #   neg.scores: scores of negative observations
  #   n_samples : number of samples to approximate AUC

  pos.sample <- sample(pos.scores, n_sample, replace=T)
  neg.sample <- sample(neg.scores, n_sample, replace=T)
  mean(1.0*(pos.sample > neg.sample) + 0.5*(pos.sample==neg.sample))
}
收藏
评论

没有任何其他软件包:

true_Y = c(1,1,1,1,2,1,2,1,2,2)
probs = c(1,0.999,0.999,0.973,0.568,0.421,0.382,0.377,0.146,0.11)

getROC_AUC = function(probs, true_Y){
    probsSort = sort(probs, decreasing = TRUE, index.return = TRUE)
    val = unlist(probsSort$x)
    idx = unlist(probsSort$ix)  

    roc_y = true_Y[idx];
    stack_x = cumsum(roc_y == 2)/sum(roc_y == 2)
    stack_y = cumsum(roc_y == 1)/sum(roc_y == 1)    

    auc = sum((stack_x[2:length(roc_y)]-stack_x[1:length(roc_y)-1])*stack_y[2:length(roc_y)])
    return(list(stack_x=stack_x, stack_y=stack_y, auc=auc))
}

aList = getROC_AUC(probs, true_Y) 

stack_x = unlist(aList$stack_x)
stack_y = unlist(aList$stack_y)
auc = unlist(aList$auc)

plot(stack_x, stack_y, type = "l", col = "blue", xlab = "False Positive Rate", ylab = "True Positive Rate", main = "ROC")
axis(1, seq(0.0,1.0,0.1))
axis(2, seq(0.0,1.0,0.1))
abline(h=seq(0.0,1.0,0.1), v=seq(0.0,1.0,0.1), col="gray", lty=3)
legend(0.7, 0.3, sprintf("%3.3f",auc), lty=c(1,1), lwd=c(2.5,2.5), col="blue", title = "AUC")

在此处输入图片说明

收藏
评论

有了pROC软件包,您可以从帮助页面使用auc()这样的函数:

> data(aSAH)
> 
> # Syntax (response, predictor):
> auc(aSAH$outcome, aSAH$s100b)
Area under the curve: 0.7314
收藏
评论

按照erik的回答,您还应该能够通过比较pos.scores和neg.scores中所有可能的值对来直接计算ROC:

score.pairs <- merge(pos.scores, neg.scores)
names(score.pairs) <- c("pos.score", "neg.score")
sum(score.pairs$pos.score > score.pairs$neg.score) / nrow(score.pairs)

肯定比示例方法或pROC :: auc效率低,但比前者更稳定,并且比后者需要更少的安装。

相关:当我尝试此操作时,它得出的结果与pROC的值相似,但并不完全相同(相差0.02左右);结果非常接近具有很高N的样本方法。如果有人知道为什么会这样,我会很感兴趣。

收藏
评论

结合ISL 9.6.3 ROC Curves中的代码以及@J。赢得了这个问题的答案以及更多其他位置,以下内容绘制了ROC曲线,并在该图的右下方打印了AUC。

下面probs是二元分类和预测概率的数字矢量test$label包含测试数据的真实标签。

require(ROCR)
require(pROC)

rocplot <- function(pred, truth, ...) {
  predob = prediction(pred, truth)
  perf = performance(predob, "tpr", "fpr")
  plot(perf, ...)
  area <- auc(truth, pred)
  area <- format(round(area, 4), nsmall = 4)
  text(x=0.8, y=0.1, labels = paste("AUC =", area))

  # the reference x=y line
  segments(x0=0, y0=0, x1=1, y1=1, col="gray", lty=2)
}

rocplot(probs, test$label, col="blue")

这给出了这样的情节:

在此处输入图片说明

收藏
评论
新手导航
  • 社区规范
  • 提出问题
  • 进行投票
  • 个人资料
  • 优化问题
  • 回答问题

关于我们

常见问题

内容许可

联系我们

@2020 AskGo
京ICP备20001863号