无法更改 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)

无法更改 ConfusionMatrix 大小

Unable to change ConfusionMatrix size

python

matplotlib

confusion-matrix

scikit-learn