【巨好用】绘制 自定义混淆矩阵

这就我见过坠好用的混淆矩阵代码！都给我用！

#confusion_matrix
import numpy as np
import matplotlib.pyplot as plt
cm_test = np.array([[2692,20],[30,707]])
classes = ['0','1']
# 关于类别顺序可由 labels参数控制调整，
# 例如 labels=[2,1,0],则类别将以这个顺序自上向下排列。默认数字类别是从小到大排列，英文类别是按首字母顺序排列
confusion_matrix = cm_test
plt.figure(figsize=(5, 4),dpi=300)
tick_marks = np.arange(len(classes))
plt.xticks(tick_marks, classes,size = 16)
plt.yticks(tick_marks, classes,size = 16)
plt.xlabel('Real label',fontsize = 18)
plt.ylabel('Prediction',fontsize = 18)
thresh = confusion_matrix.max() / 2.
iters = np.reshape([[[i,j] for j in range(2)] for i in range(2)],(confusion_matrix.size,2))
for i, j in iters:
    plt.text(j, i, format(confusion_matrix[i, j]),fontsize = 16,va = 'center', ha = 'center')   #显示对应的数字

plt.tight_layout()

# plt.title('CM')
plt.imshow(confusion_matrix, interpolation='nearest', cmap=plt.cm.GnBu)  #按照像素显示出矩阵
plt.colorbar()

整个就是一字大好看！
画布设置
标签，刻度 字体设置，
混淆矩阵字体设置（包括大小和居中）

plt.text #矩阵快内的字体在这里修改

把

cm_test = np.array([[2692,20],[30,707]])

的变量换成自己的，就可以随意定制。
元素个数，按格式调整一下就行。

cmap=plt.cm.GnBu

这里设置颜色，颜色真的很多，参考汇总：
python 画图colorbar 颜色大全 plt.cm.get_cmap

over！