使用SHAP值的XGBoost R中的全局功能重要性

时间:2019-02-11 16:11:31

标签: r xgboost

我想知道在XGBoost的R包中是否有一种使用SHAP值而不是像SHAP的Python包那样的GAIN来计算全局特征重要性的方法。 我的意思是,在用于Python的XGBoost中,有一个函数可以在全局级别上计算SHAP值,从而使每个功能的SHAP值的平均值绝对。

1 个答案:

答案 0 :(得分:0)

您可以这样做:

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

以便直接从SHAP获取XGBoost的值。

一经了解,就可以解决:

mean_shap <- colMeans(abs(shap_contrib))