ModelOriented / shapviz

SHAP Plots in R
https://modeloriented.github.io/shapviz/
GNU General Public License v2.0
79 stars 14 forks source link

XGBoost 2 #143

Closed mayer79 closed 3 months ago

mayer79 commented 3 months ago

XGBoost 2 will bring a lot of cool features. But we need to adapt the shapviz() wrapper around predict(...,):

  1. The column name "BIAS" was renamed to "(Intercept)". Since it is always the last column, we might work with integer position instead of names.

  2. In multi-output settings, instead of a list of SHAP (interaction) values, an array of one order higher is returned. The new class dimension is at position 2.

  3. The new strict_shape = TRUE option correctly handles the single row case, where the old behaviour dropped a dimension.

These proposals should work with XGBoost 1 and 2.