混淆矩阵可视化

---

一、导入模块

import pandas as pd
import matplotlib.pyplot as plt
import seaborn as sns

二、设置样式

sns.set()  #设置默认样式
sns.set_style('whitegrid',{'font.sans-serif':['simhei','Arial']})  #可显示中文

三、创建数据

df = pd.DataFrame([[1496, 12], 
              [7,118]],
              index = ['low-risk','high-risk'],
              columns = ['low-risk','high-risk'])

四、绘图保存

#图片大小
plt.figure(figsize = (10,8))
f,ax= plt.subplots(figsize = (10, 8))
#绘制
h = sns.heatmap(df,annot = True,fmt = 'd',cmap = 'Greens',linewidths=4,annot_kws={'size':20},cbar=False)
#colorbar
cb = h.figure.colorbar(h.collections[0])    #显示colorbar
cb.ax.tick_params(labelsize=20)             #设置colorbar刻度字体大小

ax.tick_params(labelsize=24)                 #标签大小
#ax.set_yticklabels(ax.get_yticklabels(), rotation=0)
ax.xaxis.set_ticks_position('top')
#坐标轴标签
plt.ylabel('True label',fontsize=24)
plt.xlabel('Predicted label',fontsize=24)
#保存图片
plt.savefig('C:/Users/HP/Desktop/混淆矩阵图.png', dpi=400,bbox_inches = 'tight')
plt.show()

---