shap / shap

A game theoretic approach to explain the output of any machine learning model.
https://shap.readthedocs.io
MIT License
22.89k stars 3.29k forks source link

custom base values for tree explainer #2934

Open nshahpazov opened 1 year ago

nshahpazov commented 1 year ago

Is there a way to set the base_values before calculating the shap values for a TreeExplainer in the shap package? I'm using a gpu xgbooster and I want to be able to set the base values.

Thanks in advance!

CloseChoice commented 11 months ago

The base values for shap are set as the average model output. Why do you want to set this value manually?