5.8 绘制混淆矩阵

1.混淆矩阵

接受者操作特征(Receiver Operating Characteristic Curve,ROC)曲线是显示分类器真正率和假正率之间折中
的⼀种图形化⽅法,也是最常⽤的评估分类模型好坏的可视化图形,在介绍ROC曲线之前,先了解下混淆矩阵。

1.1 原理

混淆矩阵是用来总结一个分类器结果的矩阵。对于k元分类,其实它就是一个k x k的表格,用来记录分类器的预测结果。 对于最常见的二元分类来说,它的混淆矩阵是2乘2的。
假设有一批test样本,这些样本只有两种类别:正例和反例。机器学习算法预测类别如下图(左半部分预测类别为正例,右半部分预测类别为反例),而样本中真实的正例类别在上半部分,下半部分为真实的反例。 如下图:
预测值为正例,记为P(Positive)
预测值为反例,记为N(Negative)
预测值与真实值相同,记为T(True)
预测值与真实值相反,记为F(False)

image.png
image.png
image.png

注: F1 score 来源:
1/ ((1/精准率)+ (1/召回率))
由于精准率,召回率最大1,此值最大0.5,会乘2,使其在0-1区间。

2. ROC曲线

image.png
image.png

横坐标:1-负样本的召回率 假正率:负样本中预测成正样本的概率
纵坐标:正样本的召回率 真正率:正样本中预测为正样本的概率

3. 绘制ROC曲线

3.1 导入波士顿房价,获取X,y

将大于均价的房价映射为1,低于映射为0,构成分类问题,并初步用逻辑斯蒂预测和评分。

import numpy as np
import pandas as pd
from pandas import Series, DataFrame

import matplotlib.pyplot as plt
%matplotlib inline

from sklearn.datasets import load_boston # 获取波士顿房价数据
X, y = load_boston(return_X_y=True) # 获取特征向量集X和目标房价 y

展示y
import seaborn as sns
sns.set()
sns.distplot(y)

image.png

根据均值将y映射为0和1
threshold = np.median(y)
target = Series(y).map(lambda x: (x>threshold)*1)
target.value_counts().plot(kind='bar')


image.png

逻辑斯蒂建模
from sklearn.linear_model import LogisticRegression
from sklearn.model_selection import train_test_split # 获取拆分函数

X_train, X_test, y_train, y_test = train_test_split(X, target, test_size=0.2, random_state=1) # 拆分样本集,拆分比例0.8:0.2,并做随机种子,下次拆分如果还用随机种子1,可以得到一样的随机结果。

预测并评分
lr = LogisticRegression()
lr.fit(X_train, y_train)
y_ = lr.predict(X_test)
y_

image.png

image.png

from sklearn.metrics import accuracy_score
accuracy_score(y_, y_test) # 0.8725490196078431
准确率评分,即预测正确的占总数的比例。如果样本比例不均,此评分高看不出来对两种预测样本的准确度。
比如:99个男生,1个女生,预测为100个男生,准确率99%,但对女生的预测率为0,所以需要引入其他评分指标。

3.2 绘制混淆矩阵

(1)导包,准备数据
from sklearn.metrics import confusion_matrix # 导入混淆矩阵的包
y_.size # 102
cm = confusion_matrix(y_test, y_)
cm # array([[44, 3], [10, 45]], dtype=int64)

tn, fp, fn, tp = confusion_matrix(y_test, y_).ravel()
(tn, fp, fn, tp) # (44, 3, 10, 45)

(2)绘图展示
sns.set_style('white') # 设置全局的图的背景为白色
def show_cm(y_true, y_test):
tn, fp, fn, tp = confusion_matrix(y_test, y_true).ravel()
plt.imshow(cm, cmap=plt.cm.Blues)
plt.xticks([0,1])
plt.yticks([0,1])
plt.xlabel("Predict")
plt.ylabel("True")

plt.text(x=0-0.1, y=0, s="TP:"+str(tp), color='black') # 设置文本,位置由x,y确定,x,y为相对位置。0-1
plt.text(x=1-0.1, y=0, s="FN:"+str(fn), color='black')
plt.text(x=0-0.1, y=1, s="FP:"+str(fp), color='black')
plt.text(x=1-0.1, y=1, s="TN:"+str(tn), >color='black')

查准率 = tp/(tp + fp)
召回率 = tp/(tp + tn)
print("查准率:",查准率)
print("召回率:",召回率)
plt.show()

绘图
show_cm(y_test, y_)


image.png

(3)对预测阀值进行修改并展示混淆矩阵
原理,逻辑斯蒂回归,根据概率判断正负样本,阀值0.5,0的概率大于1时,预测为0.

image.png

对正样本的预测阈值进行修改,查看变换
y1_ = (lr.predict_proba(X_test)[:,1] > 0.7)*1 #通过predict_proba获取每个预测值的概率,如果第2列的概率大于0.7,则为1,小于为0. 即增大正样本的概率判断的阈值。

image.png
image.png

3.3 f1_score评分

image.png

3.4 绘制 ROC曲线

3.4.1计算评分

image.png
image.png

绘制曲线
from sklearn import metrics

def show_roc(y_test, scores, pos_label):
fpr, tpr, thresholds = metrics.roc_curve(y_test, scores, pos_label=pos_label)
plt.plot(fpr, tpr, color='green')
plt.plot(np.linspace(0,1,10), np.linspace(0,1,10), color='red', ls='--')
plt.xlabel("FPR")
plt.ylabel("TPR")
plt.title("LR ROC ")
plt.show()

绘制ROC曲线

image.png

绘制不同情况下的ROC曲线
参数c更小

image.png

高斯分布朴素贝叶斯预测

from sklearn.naive_bayes import GaussianNB
gnb = GaussianNB()
gnb.fit(X_train, y_train)
scores3 = gnb.predict_proba(X_test)[:,1]
show_roc(y_test, scores3, 1)

image.png

3.5 绘制ROC曲线

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)

fpr, tpr, thresholds # 列方向看,当阈值为1,8时,预测FPR和TPR的值都为0,当阈值为0.8时,FPR为0,TPR为0.5.

(array([0. , 0. , 0.5, 0.5, 1. ]),
array([0. , 0.5, 0.5, 1. , 1. ]),
array([1.8 , 0.8 , 0.4 , 0.35, 0.1 ]))

import matplotlib.pyplot as plt
%matplotlib inline
plt.plot(fpr, tpr, color='blue', lw=3)
plt.xlabel('FPR')
plt.ylabel('TPR')

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

推荐阅读更多精彩内容