从 shapely 库中获取变量列表

【问题标题】：Get the variables list from shapely library从 shapely 库中获取变量列表
【发布时间】：2020-02-21 01:38:29
【问题描述】：

我正在使用 shap 库来获取变量重要性我遇到的问题是我只能找到绘图选项，但我需要按顺序获取变量重要性列表。

【问题讨论】：

【解决方案1】：

每个*Explainer 类都有方法shap_values，它返回一个MxN numpy 数组的列表，其中M = # objects、N = # features。

变量重要性由mean(|SHAP value|) 衡量，如repo's README 中所述。因此，如果您的数据位于 pandas 数据框中，以下是获取按重要性排序的特征列表的方法：

per_feature_mean_shap = np.abs(shap_values[0]).mean(axis=0)
df.columns[np.argsort(per_feature_mean_shap)]

【讨论】：