Rohan Shad, MD
Rohan Shad, MD
Perfect! I'll take a crack at this soon (hopefully within this week) and circle back. Thanks 👍
Got it all working on a new conda environment, and I've successfully ported the example to torch lightning. I set up the dataset within a lightning [DataModule](https://pytorch-lightning.readthedocs.io/en/stable/datamodules.html) and packaged the...
#66 Here you go ^
Let me know if you'd like to create an example for CoxPH, the workings and estimators seem to be a bit different from the logistic_hazards models. I can carry on...