如何更改 sklearn.metrics 包中的 plot_confusion_matrix 默认图形大小答案

【问题标题】：How to change plot_confusion_matrix default figure size in sklearn.metrics package如何更改 sklearn.metrics 包中的 plot_confusion_matrix 默认图形大小
【发布时间】：2020-08-03 02:58:29
【问题描述】：

我尝试使用 sklearn.metrics.plot_confusion_matrix 包使用 Jupyter notebook 绘制混淆矩阵，但默认图形大小有点小。我在绘图之前添加了 plt.figure(figsize=(20, 20)) ，但是图形大小没有随着输出文本“图形大小 1440x1440 和 0 轴”而改变。如何更改图形大小？

%matplotlib inline
from sklearn.ensemble import GradientBoostingClassifier
from sklearn.metrics import plot_confusion_matrix
from matplotlib import pyplot as plt

plt.figure(figsize=(20, 20))
clf = GradientBoostingClassifier(random_state=42)
clf.fit(X_train, y_train)
plot_confusion_matrix(clf, X_test, y_test, cmap=plt.cm.Blues)
plt.title('Confusion matrix')
plt.show()

just like this image

【问题讨论】：

fig, ax = plt.subplots(figsize=(20, 20))，然后是plot_confusion_matrix(clf, X_test, y_test, cmap=plt.cm.Blues, ax=ax)。 docs 很有帮助。

标签： python-3.x matplotlib scikit-learn jupyter-notebook confusion-matrix

【解决方案1】：

我不知道为什么 BigBen 将它作为评论而不是答案发布，但我几乎错过了看到它。这是一个答案，所以未来的围观者不会犯我几乎犯的同样错误！

fig, ax = plt.subplots(figsize=(10, 10))
plot_confusion_matrix(your_model, X_test, y_test, ax=ax)

【讨论】：