Важность глобальной функции в XGBoost R с использованием значений SHAP

Question

Я хотел бы знать, есть ли метод для вычисления важности глобальной функции в R-пакете XGBoost, используя значения SHAP вместо GAIN, как Python-пакет SHAP.Я имею в виду, что в XGBoost для Python есть функция для вычисления значений SHAP на глобальном уровне, делающая среднее абсолютное значение SHAP для каждой функции.

Answer 1

Вы можете сделать:

shap_contrib <- predict(model,newdata,predcontrib=T)

, чтобы получить SHAP значения непосредственно из XGBoost.

Как только вы это получите, это просто вопрос действий:

mean_shap <- colMeans(abs(shap_contrib))