记录一下关于python使用sklearn中的混淆矩阵库ConfusionMatrixDisplay，在图中添加title，修改label等的方法
本科没有写什么，研究生期间第一篇记录
由于
ConfusionMatrixDisplay
库会隐式调用
plt.subplots()
，导致用户不能获取到figure或者ax不能使用内部函数进行修改，通过观察源码，我们发现
ConfusionMatrixDisplay
类中有一个
plot函数
（这也是我们绘制混淆矩阵所需要调用的），函数的参数表有
ax
变量。
相信有些朋友已经知道如何解决了。
是的
通过传入
外部的ax
，使用
外部的ax
添加其他内容即可,代码如下：
confusion_matrix_figure=ConfusionMatrixDisplay(confusion_matrix=cm,display_labels=[0,1])
ax=plt.figure().subplots()
accuracy = (cm[0, 0] + cm[1, 1]) * 1.0 / np.sum(cm)
ax.set(title="Accuracy = %0.2f" % accuracy)
confusion_matrix_figure.plot(ax=ax)
plt.show()
版权声明：本文为Lz0113原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。