如何更改shap.plot.bar和shap.plots.falls中条形图的颜色

Question

正如问题中提到的，我们是否可以在shap图中更改条形图和瀑布图的颜色？我没有看到任何改变它的选项。我尝试使用matplotlib来改变颜色，但这并没有解决问题。我可以添加标题，但不能更改颜色。任何帮助都会很棒。

plt.title("This is my title")
plt.gray()
shap.plots.bar(shap_values,show = False)
shap.plots.waterfall(shap_values[-1])
plt.show()

谢谢，

Sam

Answer 1

我也遇到了类似的问题。我想在shap瀑布图中把蓝色换成红色。

我的方法是修改shap-lib中的文件_waterfall.py，将c = colors.red_rgb的行替换为c = colors.blue_rgb，反之亦然。

当然不是一个优雅的解决方案，但我只需要用这种方式生成6个图。