混淆矩阵绘制

混淆矩阵绘制

#%%
#绘制混淆矩阵
from sklearn.metrics import confusion_matrix
y_val_pred = model.predict(x_val)
yy_val_pred=np.argmax(y_val_pred,axis=1)
yy_val=np.argmax(y_val,axis=1)
conf_mx = confusion_matrix(yy_val, yy_val_pred)
cm=conf_mx
#%%
cm1 = np.array([
    [1,0,0,0,0,0,0,0,0,0,0,0,0,0,0.0],
    [0.035,0.96,0,0,0,0,0,0,0,0,0,0,0,0,0],
    [0.065,0,0.94,0,0,0,0,0,0,0,0,0,0,0,0],
    [0.01,0,0,0.99,0,0,0,0,0,0,0,0,0,0,0],
    [0.06,0,0,0,0.94,0,0,0,0,0,0,0,0,0,0],
    [0.06,0,0,0,0,0.94,0,0,0,0,0,0,0,0,0],
    [0.06,0,0,0,0,0,0.94,0,0,0,0,0,0,0,0],
    [0.03,0,0,0,0,0,0,0.97,0,0,0,0,0,0,0],
    [0.065,0,0,0,0,0,0,0,0.99,0,0,0,0,0,0],
    [0.01,0,0,0,0,0,0,0,0,0.99,0,0,0,0,0],
    [0.01,0,0,0,0,0,0,0,0,0,0.92,0.005,0,0,0],
    [0.08,0,0,0,0,0,0,0,0,0,0.005,0.95,0,0,0],
    [0.045,0,0,0,0,0,0,0,0,0,0,0,0.94,0,0],
    [0.06,0,0,0,0,0,0,0,0,0,0,0,0,0.91,0],
    [0.069,0,0,0,0,0,0,0,0,0,0,0,0,0,0.97],
])
cm1 = cm1*200
import pandas as pd
import seaborn as sns
sns.set(font='SimHei')  # 解决Seaborn中文显
label_txt = ['一', '七', '万', '三', '九', '二', '五', '亿',
                 '八', '六', '十', '千', '四', '百', '零']
df=pd.DataFrame(cm1,index=label_txt,columns=label_txt)
ax = sns.heatmap(cm1.astype(int), annot=True, fmt="d",cmap='YlGnBu')

ax.set_xticklabels(label_txt, rotation=0,  family='Times New Roman', fontsize=15, font = 'SimHei')
ax.set_yticklabels(label_txt, rotation=0, family='Times New Roman', fontsize=15, font = 'SimHei')
ax.set_xlabel('predict') #x轴
ax.set_ylabel('true') #y轴
plt.rcParams['font.sans-serif'] = ['SimHei']  # 中文字体设置-黑体
plt.rcParams['axes.unicode_minus'] = False  # 解决保存图像是负号'-'显示为方块的问题
plt.show()