簡體   English   中英

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

[英]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.

 
粵ICP備18138465號  © 2020-2024 STACKOOM.COM