Как изменить размер фигуры по умолчанию plot_confusion_matrix в пакете sklearn.metrics - PullRequest
0 голосов
/ 20 апреля 2020

Я пытался построить матрицу путаницы с ноутбуком Jupyter, используя пакет sklearn.metrics.plot_confusion_matrix, но размер фигуры по умолчанию немного мал. Я добавил 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()

так же, как это изображение

...