np.fill_diagonal(conf_mx, 0) sns.heatmap(conf_mx,annot=True,fmt="d")
时间: 2023-06-18 13:02:21 浏览: 131
半径和颜色的HeatMap
这段代码使用了NumPy库的`fill_diagonal`函数,将混淆矩阵(confusion matrix)的对角线元素设为0。然后使用Seaborn库的`heatmap`函数生成一个热力图,其中`annot=True`表示显示每个单元格的数值,`fmt="d"`表示数值以整数形式显示。这样可以更直观地观察模型预测结果与真实标签之间的差别。
阅读全文