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
78 stars 22 forks source link

Support for other survival tree-based models #31

Open krzyzinskim opened 9 months ago

krzyzinskim commented 9 months ago

Adding support for survival models other than ranger (see #24). The source code for the ranger can be followed. Functionality for randomForestSRC should be relatively easy to implement.

mayer79 commented 9 months ago

@krzyzinskim If I recall correctly, {randomForestSRC} even supports multi-output regressions (which is nice).

krzyzinskim commented 9 months ago

Hmm, in that case, I'll try to rethink the structure for handling non-single-number predictions so that multi-output regression can be also supported.

mayer79 commented 9 months ago

(Dito for XGBoost 2.0). The output structure can be like classification, I'd say: a list of matrices/arrays (?)