无法更改ConfusionMatrix的大小

Question

我有以下非常简单的python代码:

from sklearn.metrics import ConfusionMatrixDisplay
# Holdout method with 2/3 training
X_train, X_test, y_train, y_test = train_test_split(self.train, self.target, test_size=0.33)

# train the k-NN
classifier = neighbors.KNeighborsClassifier(k)
classifier.fit(X_train, y_train)
# predict the test set on our trained classifier
y_test_predicted = classifier.predict(X_test)

ConfusionMatrixDisplay.from_predictions(y_test,y_test_predicted)

事情是我的预测有几个类和绘制的矩阵只是方式小正确显示结果…(它只在Jupyter笔记本中自动绘制这一行)

我的问题很简单。如何使用这种方法增加混淆矩阵的大小?

我真的已经搜索了几十个网站，但没有任何线索。

它们似乎都以某种方式涉及到matplotlib，我只是不够聪明，无法理解它。

我现在很绝望，因为这个问题看起来很简单，但又很不可能。

如何增加绘制的混淆矩阵的大小?

Answer 1

您可以创建一个您想要的大小的ax(在下面的示例中，我将其设置为(50,50)并将其作为参数ax传递给函数)?

f,ax = plt.subplots(1,1,figsize=(50,50))
ConfusionMatrixDisplay.from_predictions(y_test, y_pred, ax=ax)