我尝试使用sklearn.metrics.plot_confusion_matrix包在Jupyter notebook中绘制混淆矩阵,但默认的图形大小有点小。我在绘图前添加了plt.figure(figsize=(20,20)),但是图形大小并没有随着输出文本'Figure size 1440x1440 with 0 Axes‘而改变。如何更改图形大小?
%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()
发布于 2020-06-29 12:54:13
我不知道为什么BigBen把它作为一个评论,而不是一个答案,但是我几乎错过了它。这是一个答案,这样以后的旁观者就不会犯我几乎犯过的同样的错误了!
fig, ax = plt.subplots(figsize=(10, 10))
plot_confusion_matrix(your_model, X_test, y_test, ax=ax)
https://stackoverflow.com/questions/61325314
复制相似问题