gbm-developers / gbm3

Gradient boosted models
134 stars 116 forks source link

Add support for SHAP value predictions #151

Open gravesee opened 6 years ago

gravesee commented 6 years ago

Recent advances in interpretability have produced SHAP values for assigning score contributions to features in a consistent manner. Calculating per-feature score contributions was added to XGBoost and LightGBM. It would be great to have it in gbm3 as well:

https://github.com/slundberg/shap