search

havakv / pycox

Survival analysis with PyTorch
BSD 2-Clause "Simplified" License
803 stars 188 forks source link

Training using PyTorch training loop #117

Open BardiaKh opened 2 years ago

BardiaKh commented 2 years ago

Thank you for developing such a useful package and also maintaining it over time. I was wondering if there is a way to use the loss functions as vanilla loss functions in a pytorch training loop? I'm asking this because I have a previously implemented pipeline for classification and wanted to see if I could integrate your packages into it. I mean bypassing wrapping the nn.Module with one of the pycox methods.

havakv commented 2 years ago

Hi, and thank you for the kind words. So the branch refactor_out_torchtuples addresses this. Hopefully, we should be able to merge this into the master branch at some point. This is also addressed in #25 and #60