如何计算部分曲线下面积(AUC)

在scikit-learn中,你可以使用以下代码计算二分类器的曲线下面积:

roc_auc_score( Y, clf.predict_proba(X)[:,1] )

我只对曲线中假阳性率小于0.1的部分感兴趣。

给定这样的假阳性率阈值,我如何计算仅限于阈值之前的曲线部分的AUC?

这里有一个包含多个ROC曲线的示例,用于说明:

不同类型分类器的ROC曲线图示。

scikit-learn的文档展示了如何使用roc_curve

>>> import numpy as np>>> from sklearn import metrics>>> y = np.array([1, 1, 2, 2])>>> scores = np.array([0.1, 0.4, 0.35, 0.8])>>> fpr, tpr, thresholds = metrics.roc_curve(y, scores, pos_label=2)>>> fprarray([ 0. ,  0.5,  0.5,  1. ])>>> tprarray([ 0.5,  0.5,  1. ,  1. ])>>> thresholdsarray([ 0.8 ,  0.4 ,  0.35,  0.1 ]

有没有简单的方法从这里计算部分AUC?


似乎唯一的问题是如何在roc_curve不一定提供的情况下计算fpr = 0.1时的tpr值。


回答:

Python sklearn的roc_auc_score()现在允许你设置max_fpr。在你的情况下,你可以设置max_fpr=0.1,该函数将为你计算AUC。 https://scikit-learn.org/stable/modules/generated/sklearn.metrics.roc_auc_score.html

Related Posts

L1-L2正则化的不同系数

我想对网络的权重同时应用L1和L2正则化。然而,我找不…

使用scikit-learn的无监督方法将列表分类成不同组别,有没有办法?

我有一系列实例,每个实例都有一份列表,代表它所遵循的不…

f1_score metric in lightgbm

我想使用自定义指标f1_score来训练一个lgb模型…

通过相关系数矩阵进行特征选择

我在测试不同的算法时,如逻辑回归、高斯朴素贝叶斯、随机…

可以将机器学习库用于流式输入和输出吗?

已关闭。此问题需要更加聚焦。目前不接受回答。 想要改进…

在TensorFlow中,queue.dequeue_up_to()方法的用途是什么?

我对这个方法感到非常困惑,特别是当我发现这个令人费解的…

发表回复

您的邮箱地址不会被公开。 必填项已用 * 标注