search

havakv / pycox

Survival analysis with PyTorch
BSD 2-Clause "Simplified" License
781 stars 180 forks source link

Pycox importance variables #104

Open Nusha34 opened 2 years ago

Nusha34 commented 2 years ago

Does pycox has function for calculating variable importance? Didn't see this function inside the code

Many thanks for your answer in advance

havakv commented 2 years ago

Sadly there is no variable importance implemented here. But I think it would be valuable, so if anyone want to have a go at it, that would be appreciated

puckybreg commented 1 year ago

@havakv how would someone go about calculating the variable importance from the trained model. It would involve interpreting the weights of the neural net correct?