python - 如何在 SHAP 汇总图上绘制特定特征？

Question

我目前正在尝试在 SHAP 汇总图上绘制一组特定功能。但是，我正在努力寻找这样做所需的代码。

查看 Github 上的源代码时，summary_plot 函数似乎确实具有“特征”属性。但是，这似乎不能解决我的问题。

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

标签： pythonplotshap

Answer 1

一个可能的解决方案虽然很老套，但可能如下所示，例如在第 5 列中为单个特征绘制摘要图

shap.summary_plot(shap_values[:,5:6], X.iloc[:, 5:6])