如何将绘图（由 shap_values 生成）保存到 png？答案

【问题标题】：How to save a plot (generated by shap_values) to png?如何将绘图（由 shap_values 生成）保存到 png？
【发布时间】：2019-09-17 14:50:32
【问题描述】：

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

我尝试将 shap_summary_plot 保存为“png”图像，但我的 image.png 但他们得到一个空图像

这是我使用的代码：

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？

【问题讨论】：

标签： python-3.x plot save png shap

【解决方案1】：

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

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')

【讨论】：