iml icon indicating copy to clipboard operation
iml copied to clipboard

Support survival analysis tasks

Open RaphaelS1 opened this issue 3 years ago • 3 comments

Hi Christoph,

Thanks for making models more interpretable! Would you be happy to start supporting survival analysis in iml (via mlr3proba)? I'm happy to make a PR for this if you are.

Thanks!

RaphaelS1 avatar Apr 19 '21 15:04 RaphaelS1

What would you use as a prediction in the case of survival analysis? Do you have a quick proof of concept using a custom predict.fun for Predictor$new()?

christophM avatar Apr 23 '21 08:04 christophM

What would you use as a prediction in the case of survival analysis?

Agreed it isn't straight forward and it is messy but in general I'd say the only possible compatible prediction is a continuous relative risk ranking which you would then compute concordance on. Has little meaning in interpreting things like SHAPs (what does a relative increase of a relative prediction even mean) but is useful for feature importance.

Do you have a quick proof of concept using a custom predict.fun for Predictor$new()?

Can't Predictor just wrap mlr models?

RaphaelS1 avatar Apr 23 '21 10:04 RaphaelS1

It would be nice to create ALE plots in addition to the pdp and marginal effects plots cabaple of being produced with these models: https://www.randomforestsrc.org/articles/partial.html

Random forest survival models

Jack-FG avatar Oct 01 '23 22:10 Jack-FG