我使用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?
发布于 2019-09-18 14:36:53
我解决了这个问题。代码是:
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')
https://stackoverflow.com/questions/57976902
复制相似问题