ConfusionMatrixDisplay,在图中添加title,修改label等的方法

记录一下关于python使用sklearn中的混淆矩阵库ConfusionMatrixDisplay,在图中添加title,修改label等的方法

  • 本科没有写什么,研究生期间第一篇记录

本科没有写什么,研究生期间第一篇记录

由于ConfusionMatrixDisplay库会隐式调用plt.subplots(),导致用户不能获取到figure或者ax不能使用内部函数进行修改,通过观察源码,我们发现ConfusionMatrixDisplay类中有一个plot函数(这也是我们绘制混淆矩阵所需要调用的),函数的参数表有ax变量。
ConfusionMatrixDisplay,在图中添加title,修改label等的方法_第1张图片
相信有些朋友已经知道如何解决了。
是的通过传入外部的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()

你可能感兴趣的:(python,机器学习可视化,python,机器学习)