我使用 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')
Run Code Online (Sandbox Code Playgroud)
代码有效,但保存的图像是空的。
如何将绘图保存为 image.png?