首页
学习
活动
专区
圈层
工具
发布
首页
学习
活动
专区
圈层
工具
MCP广场
社区首页 >问答首页 >如何将绘图(由shap_values生成)保存为png?

如何将绘图(由shap_values生成)保存为png?
EN

Stack Overflow用户
提问于 2019-09-17 22:50:32
回答 1查看 2.8K关注 0票数 1

我使用Shap库来可视化变量的重要性。

我尝试将shap_summary_plot另存为'png‘图像,但我的image.png得到一个空图像

这是我使用的代码:

代码语言:javascript
运行
复制
shap_values = shap.TreeExplainer(modelo).shap_values(X_train)
shap.summary_plot(shap_values, X_train, plot_type="bar")
plt.savefig('grafico.png')

代码起作用了,但是保存的图像是空的。

如何将绘图另存为image.png?

EN

回答 1

Stack Overflow用户

发布于 2019-09-18 14:36:53

我解决了这个问题。代码是:

代码语言:javascript
运行
复制
shap_values = shap.TreeExplainer(modelo).shap_values(X_train)
shap.summary_plot(shap_values, X_train, plot_type="bar",show=False)
plt.savefig('grafic.png')
票数 4
EN
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/57976902

复制
相关文章

相似问题

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