Получить список переменных из стройной библиотеки

Question

Я использую библиотеку shap для получения важности переменных. Проблема в том, что я нахожу только параметры графиков, но мне нужно получить список важности переменных по порядку.

Answer 1

Каждый *Explainer класс имеет метод shap_values, который возвращает список массивов MxN numpy, где M = # objects, N = # features.

Важность переменной измеряется mean(|SHAP value|), как указано в README РЕПО . Поэтому, если ваши данные находятся в pandas фрейме данных, вот как получить список отсортированных по важности функций:

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