混淆矩阵

简介

混淆矩阵是衡量分类模型准确度中最基本、直观,计算最简单的方法。混淆矩阵就是一张表,展示预测值各类归对、归错的个数。下面放个小例子。

小例子

上面的小例子是分类问题中最简单的二分类,只需要判断结果是0还是1,或者说是negative还是positive。

表中已经给出四种结果,这四种是混淆矩阵的四个基础指标:

  • 真实值是Positive,模型认为是Positive的个数(True Positive = TP
  • 真实值是Positive,模型认为是Negative的个数(False Negative = FN),统计学上的Type I Error(去真)
  • 真实值是Negative,模型认为是Positive的个数(False Positive = FP),统计学上的Type II Error(存伪)
  • 真实值是Negative,模型认为是Negative的个数(True Negative = TN

将这四个指标一起呈现在表格中,就能得到混淆矩阵(Confusion Matrix),如下图。

混淆矩阵

sklearn中混淆矩阵

sklearn.metrics.confusion_matrix(y_true, y_pred, labels=None, sample_weight=None)

函数功能:

计算混淆矩阵以评估分类器的准确性。

在混淆矩阵C中,C_{i,j}的含义是实际上是i类但被错误预测为j类的样本个数。

所以在二分类中,True Negative的个数是C_{0,0},False Negative的个数是C_{1,0},True Positive的个数是C_{1,1},False Positive的个数是C_{0,1}

参数

  • y_true : array, shape = [n_samples]

    实际目标结果,真实值

  • y_pred : array, shape = [n_samples]

    分类器返回的预测值

  • labels : array, shape = [n_classes], optional

    矩阵索引的标签列表。这可用于重新排序或选择标签的子集。如果没有给值,那么在y_true或y_pred中至少出现一次的那些值按排序顺序使用。

  • sample_weight : array-like of shape = [n_samples], optional

    样本权重

返回值

  • C : array, shape = [n_classes, n_classes]

    混淆矩阵

示例

>>> y_true = [2, 0, 2, 2, 0, 1]
>>> y_pred = [0, 0, 2, 2, 0, 2]
>>> confusion_matrix(y_true, y_pred)
array([[2, 0, 0],
       [0, 0, 1],
       [1, 0, 2]])
>>> y_true = ["cat", "ant", "cat", "cat", "ant", "bird"]
>>> y_pred = ["ant", "ant", "cat", "cat", "ant", "cat"]
>>> confusion_matrix(y_true, y_pred, labels = ['bird','ant','cat'])
array([[0, 0, 1],
       [0, 2, 0],
       [0, 1, 2]])

在二分类问题中,我们可以提取True Positive等值,如下所示。

>>> tn, fp, fn, tp = confusion_matrix([0, 1, 0, 1], [1, 1, 1, 0]).ravel()
>>> (tn, fp, fn, tp)
(0, 2, 1, 1)

衍生指标

  • 准确率(Accuracy)——针对整个模型
  • 精确率(Precision)
  • 灵敏度(Sensitivity)——召回率(Recall)
  • 特异度(Specificity)

其中,Precision和Recall可以看下面的具体例子。

四种指标的公式和含义见下表。

公式 含义
准确率ACC Accuracy = \frac{TP + TN}{TP + TN + FP + FN} 所有预测正确的结果占总观测值的比例
精确率PPV Precision = \frac{TP}{TP + FP} 所有判断为Positive的结果中,分类正确的比例
灵敏度TPR Recall = \frac{TP}{TP + FN} 在所有真实值为Positive的结果中,分类正确的比例
特异度TNR Specificity = \frac{TN}{TN + FP} 在所有真实值为Negative的结果中,分类正确的比例

其中,将PrecisionRecall组合计算可以得到一个称为F1 score 的指标,如果召回和精确度都很高,分类器将获得高F 1分数。

F_1 = 2\over {1\over precision}+{1\over recall} = 2× precision×recall\over precision+recall = TP\over {TP}+{FN + FP\over 2}

代码示例

>>> from sklearn.metrics import precision_score, recall_score, f1_score
>>> y_true = [0, 0, 1, 0, 1, 0]
>>> y_pred = [0, 1, 0, 1, 1, 0]
>>> precision_score(y_true, y_pred)
0.3333333333333333
>>> recall_score(y_true, y_pred)
0.5
>>> f1_score(y_true, y_pred)
0.4

都看到最后了,要不~点个赞?加波关注?

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 215,463评论 6 497
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 91,868评论 3 391
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 161,213评论 0 351
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 57,666评论 1 290
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 66,759评论 6 388
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 50,725评论 1 294
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 39,716评论 3 415
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 38,484评论 0 270
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 44,928评论 1 307
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 37,233评论 2 331
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 39,393评论 1 345
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 35,073评论 5 340
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 40,718评论 3 324
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 31,308评论 0 21
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 32,538评论 1 268
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 47,338评论 2 368
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 44,260评论 2 352

推荐阅读更多精彩内容