如何在 SHAP 汇总图上绘制特定特征?

Vin*_*ier 4 python plot shap

我目前正在尝试在 SHAP 摘要图上绘制一组特定特征。但是,我正在努力寻找执行此操作所需的代码。

当查看 Github 上的源代码时,summary_plot 函数似乎确实有一个“features”属性。然而,这似乎并不能解决我的问题。

任何人都可以帮我绘制一组特定的功能吗?或者这在当前的 SHAP 代码中不是一个可行的选项。

小智 6

一个可能的解决方案(尽管有点hacky)可能如下,例如在第五列中绘制单个特征的摘要图

shap.summary_plot(shap_values[:,5:6], X.iloc[:, 5:6])
Run Code Online (Sandbox Code Playgroud)