我正在使用 shap 库来获取变量重要性我遇到的问题是我只找到绘图选项，但我需要按顺序获取变量重要性列表。
问问题
166 次
1 回答
0
每个*Explainer类都有shap_values返回 MxN numpy 数组列表的方法，其中M = # objects, N = # features.
变量重要性的衡量标准mean(|SHAP value|)如repo 的 README中所述。因此，如果您的数据位于 pandas 数据框中，以下是获取按重要性排序的特征列表的方法：
per_feature_mean_shap = np.abs(shap_values[0]).mean(axis=0)
df.columns[np.argsort(per_feature_mean_shap)]
于 2020-02-26T13:56:44.777 回答