我想知道是否有一种方法可以使用SHAP值而不是像SHAP的Python包这样的GAIN来计算XGBoost的R包中的全局特征重要性。我的意思是,在XGBoost for Python中,有一个函数可以在全局级别计算SHAP值,从而得到每个特征的SHAP值的平均值。
你可以做:
shap_contrib <- predict(model,newdata,predcontrib=T)
为了直接从SHAP
获得XGBoost
值。
一旦你得到它,这只是一个问题:
mean_shap <- colMeans(abs(shap_contrib))