shap.force_plot()保存其生成的图片遇到的bug

第一次写博客,有错的地方还请指正,感谢!

做毕设需要保存shap.force_plot()生成的图片,但是plt.savefig()保存为空白,后来去问学长,学长说查看他们的源代码。后反复尝试,shap.force_plot()也是内置的matplotlib,所以plt.savefig()仍然适用,但是需要传入参数matplotlib=True。

另外传入之后保存仍然是空白图片!需要将里面内置函数show()改为false,后面就顺利运行啦!

代码如下:

shap.force_plot(explainer_1.expected_value[0], shap_values_1[0], data_arry, matplotlib=True,show=False)
plt.savefig('存储路径/RR.jpg')

踩过的坑,希望能帮助到大家,谢谢!




你可能感兴趣的:(shap,大数据)