cm = confusion_matrix(y_test, y_pred) plt.figure(figsize = (8,8)) sns.heatmap(cm,cmap= "Blues", linecolor = 'black' , linewidth = 1 , annot = True, fmt='' , xticklabels = ['A','B','C','D'] , yticklabels = ['A','B','C','D']) plt.xlabel("Predicted") plt.ylabel("Actual")对生成的图片进行解释
时间: 2024-01-10 14:02:24 浏览: 22
Classifikation_regularization
这段代码用于绘制混淆矩阵的热力图,并且添加了标签和注释。其中,混淆矩阵cm是通过输入真实标签y_test和模型预测标签y_pred计算得到的。绘制的热力图大小为8*8,颜色使用蓝色调色板,边框线颜色为黑色,线宽为1,标注了每个方格的值。x轴和y轴分别表示预测值和真实值的标签,这里分别为A、B、C、D。这个热力图可以帮助我们更直观地理解分类模型的性能,进而进行模型调整和优化。
阅读全文