sns.heatmap(cf_matrix, annot=True, xticklabels = sorted(set(y_test)), yticklabels = sorted(set(y_test)))
时间: 2023-06-17 18:03:12 浏览: 56
3d_heatmap-master.zip_3D heatmap_python 3D热力图_python热力图_quietly1
5星 · 资源好评率100%
这行代码是用来生成混淆矩阵热力图的。其中,cf_matrix是混淆矩阵,annot=True表示在热力图上显示具体数值,xticklabels和yticklabels是用于设置x轴和y轴的标签,sorted(set(y_test))是将y_test中的标签去重并按照字母顺序排序。热力图可以直观地展示模型分类的准确性和错误情况。
阅读全文