How to change plot_confusion_matrix default figure size in sklearn.metrics package
11,353
I don't know why BigBen posted that as a comment, rather than an answer, but I almost missed seeing it. Here it is as an answer, so future onlookers don't make the same mistake I almost made!
fig, ax = plt.subplots(figsize=(10, 10))
plot_confusion_matrix(your_model, X_test, y_test, ax=ax)
Author by
Admin
Updated on July 24, 2022Comments
-
Admin almost 2 years
I tried to plot confusion matrix with Jupyter notebook using sklearn.metrics.plot_confusion_matrix package, but the default figure size is a little bit small. I have added plt.figure(figsize=(20, 20)) before plotting, but the figure size did not change with output text 'Figure size 1440x1440 with 0 Axes'. How can I change the figure size?
%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()