调整 ConfusionMatrixDisplay 的大小（Scikit Learn）

【问题标题】：Adjust size of ConfusionMatrixDisplay (ScikitLearn)调整 ConfusionMatrixDisplay 的大小（Scikit Learn）
【发布时间】：2021-06-03 14:16:34
【问题描述】：

如何设置 ScikitLearn 的 ConfusionMatrixDisplay 绘制的图形的大小？

import numpy as np
from sklearn.metrics import ConfusionMatrixDisplay, confusion_matrix
cm = confusion_matrix(np.arange(25), np.arange(25))
cmp = ConfusionMatrixDisplay(cm, display_labels=np.arange(25))
cmp.plot()

上面的代码就是这个图，太紧了：

【问题讨论】：

标签： python matplotlib scikit-learn data-science confusion-matrix

【解决方案1】：

您可以将matplotlib.axes 对象发送到sklearn.metrics.ConfusionMatrixDisplay 的.plot 方法。先在matplotlib.pyplot.subplots中设置图形的大小。

import numpy as np
from sklearn.metrics import ConfusionMatrixDisplay, confusion_matrix
import matplotlib.pyplot as plt

cm = confusion_matrix(np.arange(25), np.arange(25))
cmp = ConfusionMatrixDisplay(cm, display_labels=np.arange(25))
fig, ax = plt.subplots(figsize=(10,10))
cmp.plot(ax=ax)

【讨论】：