python-3.x - 如何将绘图（由 shap_values 生成）保存到 png？

Question

我使用 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？

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