【Pytorch-从一团乱麻到入门】：4、模型效果评估指标：ROC-AUC、PR-AUC及可能遇到的问题(1)

60 阅读 0 评论 40 点赞

我是靠谱客的博主轻松奇迹，最近开发中收集的这篇文章主要介绍【Pytorch-从一团乱麻到入门】：4、模型效果评估指标：ROC-AUC、PR-AUC及可能遇到的问题(1)，觉得挺不错的，现在分享给大家，希望可以做个参考。

概述

1.ROC-AUC 和PR-AUC定义

AUC：随机抽出一对样本（一个正样本，一个负样本），然后用训练得到的分类器来对这两个样本进行预测，预测得到正样本的概率大于负样本概率的概率。

ROC-AUC 指的是 ROC 曲线下的面积，通过在【0，1】范围内设置阈值来计算对应的TPR和FPR，最终将所有的点连起来构成ROC曲线。

PR-AUC 的构造和上述过程基本一致，只是需要再计算出 Precision 和 Recall ,以precision（精准率）和recall（召回率）这两个为变量而做出的曲线，其中recall为横坐标，precision为纵坐标。设定一系列阈值，计算每个阈值对应的recall和precision，即可计算出PR曲线各个点。

关于AUC的计算可参考： (5条消息) AUC的计算方法_SCUT_Sam-CSDN博客_auc计算公式

2.ROC-AUC 和PR-AUC的使用场景

所以当类别相对来说较均衡时，可以使用 ROC-AUC，当类别极其不均衡时使用 PR-AUC 较好。

从各自两个指标来看，TPR 和 FPR 分别聚焦于模型对正样本和负样本的分类能力，而 Precision 和 Recall 都是针对正样本的指标，没有考虑负样本。所以当我们希望模型在正负样本上都能表现较好时使用 ROC-AUC 衡量，如果我们只关注模型对正样本的分辨能力使用 PR-AUC 更好

3.程序代码如何实现RP-AUC、ROC-AUC的计算，使用sklearn库。

####可以使用sklearn.metrics.precision_recall_curve 来计算PR曲线

from sklearn.metrics import precision_recall_curve,auc,accuracy_score
import numpy as np
from sklearn import metrics

label=[1,0,0,0,1,0]
prob=[1,0,0,1,0,0]

precision, recall, thresholds = precision_recall_curve(ture_lab,pre_lab)

###使用AUC函数计算出PR-auc值
auc_precision_recall = auc(recall, precision)

####'''计算ROC-AUC值'''#####
roc_auc=metrics.roc_auc_score(label,prob)

4.遇到的问题

1）ROC-AUC 计算时，如果遇到“ValueError: Only one class present in y_true. ROC AUC score is not defined in that case.”问题：

可能是因为label只有一种类型导致的，数据过分不平衡，针对这种情况：

如果是用于模型训练中的测试集，要在脚本中事先做好label只有一类情况的定义

如果是用于模型验证中的验证集，也是要实现做好实际类别是否只有一类的判断，然后做相应的计算。

报错示例如下：

label=[1,1,1,1,1,1]
prob=[0,0,0,0,0,0]

报错如下：

解决此问题的方法：

避免数据不平衡情况，或者在代码中加入判断行，可以使用try ：

import numpy as np
from sklearn.metrics import roc_auc_score
label=[1,1,1,1,1,1]
prob=[0,0,0,0,0,0]
try:
roc_auc=roc_auc_score(label, prob)
except ValueError:
pass ##或者其它定义，例如roc_auc=0

2）计算PR-ROC时，隐藏bug，如果label中只有一类的数据，计算PR-RIC时并不报错，但是会计算错误

示例如下：

label=[1,1,1,1,1,1]
prob=[0,0,0,0,0,0]
'''计算PR曲线的AUC值'''
precision, recall, _thresholds = metrics.precision_recall_curve(label, prob)
print("recall","precison:",recall,precision)
print("PR-AUC:",metrics.auc(recall, precision))
print("label:",label)
print("prob",prob)

错误结果示例：