r - 如何计算 R 中子样本的 AUC

Question

我设法为我的数据集计算了一个整体 AUC，但现在我有兴趣计算和比较我的数据集子样本的 AUC。关于如何在 R 中解决这个问题的任何想法？

好心，IG

标签： rpredictionauc

Answer 1

如果没有一些数据样本，很难为您提供帮助，但让我们假设您有一个包含一些 id、一些预测和一些结果的数据框。

然后，您可以使用 dplyr 对数据进行分组并对数据子集执行计算。我使用包 pROC 来计算组的 auc。

d <- tibble::tibble(
  id = c(1,1,1,1,1,2,2,2,2,2),
  predicitons = runif(10),
  outcome = factor(c("yes", "no", "yes", "no", "no", "no", "yes", "no", "yes", "yes")))

d %>% 
  dplyr::group_by(id) %>% 
  dplyr::summarise(
    auc = as.numeric(pROC::auc(predictor = predicitons, response = outcome)))

祝你好运！