Model Interpretability for Survival task in MLR3 #980
Unanswered
FerialHantash
asked this question in
Q&A
Replies: 0 comments
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
-
Hello,
I have performed previously model interpretation (pdp) on my survival models in MLR.
However, I am unable to do it in MLR3 due to the "Predictor" object that is not accept models from type survival.
I have attached below a sample code from MLR, is there anyways to do model interpretation in MLR3?
"mod = train(lrn,surv.task)
task.pred = predict(mod,newdata = traindata)
getLearnerModel(mod)
pd = generatePartialDependenceData(mod,surv.task,c("X","Y","Z"))
plotPartialDependence(pd)"
as for c("X","Y","Z")) #These are three random features (I want to be able to do it for all features and for selected ones like in MLR)
Beta Was this translation helpful? Give feedback.
All reactions