混淆矩阵的最简单的可视化方法

对于混淆矩阵的可视化,网上大部分都是用matplot进行绘制的,效果如下混淆矩阵的最简单的可视化方法_第1张图片

然而针对这种使用matplot代码的方法,网上找了好几个,都没有绘制成功,而且代码也有点杂。可以使用seaborn进行绘制,简单方便。直接上效果图混淆矩阵的最简单的可视化方法_第2张图片

贴出代码:

import seaborn as sn
#得到混淆矩阵
cm = confusion_matrix(y_test_undersample,y_pred_undersample,labels=[0,1])

cm
#转换成dataframe,转不转一样
df_cm = pd.DataFrame(cm)
df_cm
#annot = True 显示数字 ,fmt参数不使用科学计数法进行显示
ax = sn.heatmap(df_cm,annot=True,fmt='.20g')
ax.set_title('confusion matrix') #标题
ax.set_xlabel('predict') #x轴
ax.set_ylabel('true') #y轴

 

 

你可能感兴趣的:(数据分析挖掘——工具使用)