创建函数,R语言求解cutoff, AUC, 95%置信区间,敏感性,特异性

cal_metrics <- function(label, pred){

  roc.p=pROC::roc(label, pred,ci =T)

  if (roc.p$auc>0.5){

    cutoff=roc.p$thresholds[which.max(roc.p$sensitivities+roc.p$specificities)]

    sensitivity=roc.p$sensitivities[which.max(roc.p$sensitivities+roc.p$specificities)]

    specificity=roc.p$specificities[which.max(roc.p$sensitivities+roc.p$specificities)]

    ci = roc.p$ci

    df=data.frame(type='positive classification',

                  auc=round(roc.p$auc,3),cutoff=cutoff,

                  sensitivity=sensitivity,specificity=specificity,ci_lower = ci[1],ci_upper=ci[3])

    return(df)

  }

  else{

    cutoff=roc.p$thresholds[which.min(roc.p$sensitivities+roc.p$specificities)]

    sensitivity=roc.p$sensitivities[which.min(roc.p$sensitivities+roc.p$specificities)]

    specificity=roc.p$specificities[which.min(roc.p$sensitivities+roc.p$specificities)]

    ci = roc.p$ci

    df=data.frame(type='negative classification',

                  auc=1-round(roc.p$auc,3),cutoff=cutoff,

                  sensitivity=1-sensitivity,specificity=1-specificity,ci_lower = ci[1],ci_upper=ci[3])

    return(df)

  }

}

©著作权归作者所有,转载或内容合作请联系作者
【社区内容提示】社区部分内容疑似由AI辅助生成,浏览时请结合常识与多方信息审慎甄别。
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

相关阅读更多精彩内容

友情链接更多精彩内容