首页
学习
活动
专区
圈层
工具
发布
首页
学习
活动
专区
圈层
工具
MCP广场
社区首页 >问答首页 >如何在sklearn.metrics包中更改plot_confusion_matrix默认图形大小

如何在sklearn.metrics包中更改plot_confusion_matrix默认图形大小
EN

Stack Overflow用户
提问于 2020-04-20 22:48:14
回答 1查看 9.7K关注 0票数 9

我尝试使用sklearn.metrics.plot_confusion_matrix包在Jupyter notebook中绘制混淆矩阵,但默认的图形大小有点小。我在绘图前添加了plt.figure(figsize=(20,20)),但是图形大小并没有随着输出文本'Figure size 1440x1440 with 0 Axes‘而改变。如何更改图形大小?

代码语言:javascript
运行
复制
%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()

just like this image

EN

回答 1

Stack Overflow用户

发布于 2020-06-29 12:54:13

我不知道为什么BigBen把它作为一个评论,而不是一个答案,但是我几乎错过了它。这是一个答案,这样以后的旁观者就不会犯我几乎犯过的同样的错误了!

代码语言:javascript
运行
复制
fig, ax = plt.subplots(figsize=(10, 10))
plot_confusion_matrix(your_model, X_test, y_test, ax=ax)
票数 15
EN
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/61325314

复制
相关文章

相似问题

领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档