matplotlib - 调整 ConfusionMatrixDisplay 的大小 (ScikitLearn)

Question

如何设置 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()

上面的代码显示了这个图，太紧了:

Answer 1

您可以发送 ax plot 的参数现有的 PyPlot 图。之前只设置图形的大小。

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)