search

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
79 stars 23 forks source link

random survival forest #19

Open mao223 opened 2 years ago

mao223 commented 2 years ago

Good day. Is it possible to get SHAP values for random survival forest models with this package? For instance, those generated from randomForestSRC? Thank you very much.

mikolajsp commented 2 years ago

Hey @mao223, thanks for showing interest in this package.

As random survival forests produce functional output it is difficult to apply the treeshap method directly, so this functionality isn't planned for now.

However, I encourage you to check out the survex package designed for explaining survival models. It implements the extension of survSHAP method for all survival models.