使用SHAP值在XGBoost R中的全局特征重要性

问题描述 投票:0回答:1

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

r xgboost
1个回答
0
投票

你可以做:

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

为了直接从SHAP获得XGBoost值。

一旦你得到它,这只是一个问题:

mean_shap <- colMeans(abs(shap_contrib))
© www.soinside.com 2019 - 2024. All rights reserved.