ModelOriented / treeshap

Compute SHAP values for your tree-based models using the TreeSHAP algorithm
https://modeloriented.github.io/treeshap/
GNU General Public License v3.0
78 stars 22 forks source link

Added ranger_surv.unify for random survival forests with {ranger} #22

Closed kapsner closed 1 year ago

kapsner commented 1 year ago

This PR introduces the function ranger_surv.unify in order to compute tree-shap values for random survival forest models built with the {ranger} 📦.

(Partly) addresses #19

Edit: The predictions for the ranger-survival models are computed using the sum over chf as explained here.

pbiecek commented 1 year ago

thanks!