《机器学习(周志华)》笔记--模型的评估与选择(6)--衡量分类任务的性能指标:ROC曲线与AUC计算与绘制

五、衡量分类任务的性能指标

  5、ROC曲线与AUC

  (1)ROC曲线

  ROC曲线( Receiver Operating Cha\fracteristic Curve )描述的 TPR ( True Positive Rate )与 FPR ( False Positive Rate )之间关系的曲线。

  TPR 与 FPR 的计算公式如下:

                                           

举例:

真实\预测 0 1
0 9978 12
1 2 8

      FPR=12/9990=0.0012

     TPR=8/10=0.8

分析:

  TPR描述的是模型预测Positive并且预测正确的数量占真实类别为Positive样本的比例。

  FPR描述的模型预测Positive并且预测错了的数量占真实类别为Negtive样本的比例。

  (2)TPR 与 FPR 关系

  假设有这么一组数据,菱形代表 Positive,圆形代表 Negtive。

                           

  现在需要训练一个逻辑回归的模型对数据进行分类,假如将从 0 到 1 中的一些值作为模型的分类阈值。若模型认为当前数据是 Positive 的概率小于分类阈值则分类为 Negtive ,否则就分类为 Positive (*假设分类阈值为 0.8,模型认为这条数据是 Positive 的概率为 0.7, 0.7 小于 0.8,那么模型就认为这条数据是 Negtive *)。在不同的分类阈值下,模型所对应的 TPR 与 FPR 如下图所示(竖线代表分类阈值,模型会将竖线左边的数据分类成 Negtive,竖线右边的分类成 Positive ):

                      

  从图中可以看出,当模型的 TPR 越高 FPR 也会越高, TPR 越低 FPR 也会越低。这与精准率和召回率之间的关系刚好相反。并且,模型的分类阈值一但改变,就有一组对应的 TPR 与 FPR

 举例:

TPR 0.20 0.35 0.37 0.51 0.53 0.56 0.71 0.82 0.92 0.93
FPR 0.08 0.10 0.111 0.12 0.13 0.14 0.21 0.26 0.41 0.42


   若将 FPR 作为横轴, TPR 作为纵轴,将上面的表格以折线图的形式画出来就是 ROC曲线 : 

                                     

  假设现在有模型 A 和模型 B ,它们的 ROC 曲线如下图所示(其中模型A的ROC曲线为黄色,模型B的 ROC 曲线为蓝色):

                                            

   那么模型 A 的性能比模型 B 的性能好,因为模型 A 当 FPR 较低时所对应的 TPR 比模型 B 的低 FPR 所对应的 TPR 更高。由由于随着 FPR 的增大, TPR 也会增大。所以 ROC 曲线与横轴所围成的面积越大,模型的分类性能就越高。而 ROC曲线的面积称为AUC

   (3)AUC

   很明显模型的 AUC 越高,模型的二分类性能就越强。AUC 的计算公式如下:

                                       

  其中 M 为真实类别为 Positive 的样本数量,N 为真实类别为 Negtive 的样本数量。ranki 代表了真实类别为 Positive 的样本点额预测概率从小到大排序后,该预测概率排在第几。

举例:

  现有预测概率与真实类别的表格如下所示(其中 0 表示 Negtive, 1 表示 Positive ):

编号 预测概率 真实类别
1 0.1 0
2 0.4 0
3 0.3 1
4 0.8 1

  想要得到公式中的 rank,就需要将预测概率从小到大排序,排序后如下:

编号 预测概率 真实类别
1 0.1 0
3 0.3 1
2 0.4 0
4 0.8 1

  排序后的表格中,真实类别为 Positive 只有编号为 3 和编号为 4 的数据,并且编号为 3 的数据排在第 2 ,编号为 4 的数据排在第 4。所以 rank=[2, 4] 。又因表格中真是类别为 Positive 的数据有 2 条, Negtive 的数据有 2 条。因此 M 为2, N 为2。所以根据 AUC 的计算公式可知:

                                                           

 绘制AUC曲线,代码实现:

import numpy as np

def calAUC(prob, labels):
    '''
    计算AUC并返回
    :param prob: 模型预测样本为Positive的概率列表,类型为ndarray
    :param labels: 样本的真实类别列表,其中1表示Positive,0表示Negtive,类型为ndarray
    :return: AUC,类型为float
    '''

    f = list(zip(prob,labels))
    #zip函数能将列表序列。依次取出组成一个由元组组成的列表
    rank = [values2 for values1,values2 in sorted(f,key = lambda x:x[0])]
    # 按概率从小到大排序
    rankList = [i+1 for i in range(len(rank)) if rank[i] == 1]
    # 得到rank
    posNum = 0
    negNum = 0
    for i in range(len(labels)):
        if(labels[i] == 1):
            posNum += 1
        else:
            negNum += 1
    # 根据公式计算AUC
    auc = (sum(rankList) - (posNum*(posNum+1))/2)/(posNum*negNum)
    return auc

猜你喜欢

转载自www.cnblogs.com/lsm-boke/p/12198909.html