[英]Global feature importance in XGBoost R using SHAP values
我想知道在XGBoost的R包中是否存在使用SHAP值而不是像SHAP的Python包那樣的GAIN來計算全局特征重要性的方法。 我的意思是,在適用於Python的XGBoost中,有一個函數可以在全局級別上計算SHAP值,從而使每個功能的SHAP值的平均值絕對相等。
你可以做:
shap_contrib <- predict(model,newdata,predcontrib=T)
為了直接從XGBoost
獲取SHAP
值。
一旦知道了,這就是要做的事情:
mean_shap <- colMeans(abs(shap_contrib))
聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.