treeshap icon indicating copy to clipboard operation
treeshap copied to clipboard

Support for other survival tree-based models

Open krzyzinskim opened this issue 2 years ago • 3 comments

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.

krzyzinskim avatar Oct 02 '23 11:10 krzyzinskim

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

mayer79 avatar Oct 02 '23 12:10 mayer79

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.

krzyzinskim avatar Oct 02 '23 12:10 krzyzinskim

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

mayer79 avatar Oct 02 '23 15:10 mayer79