auton-survival icon indicating copy to clipboard operation
auton-survival copied to clipboard

Add usage example notebooks

Open chiragnagpal opened this issue 5 years ago • 7 comments
trafficstars

Currently we do not have notebooks that compare performance of DSM with other models.

We would need to compare against: -DeepHit -DeepSurv -Random Survival Forest -Cox PH on Time Dependent CI and Brier Score.

chiragnagpal avatar Oct 29 '20 08:10 chiragnagpal

@chiragnagpal sir, This looks interesting... I would like to work on this part...

kishanmaharaj avatar Oct 29 '20 12:10 kishanmaharaj

@kishanmaharaj great, I am assigning this to you. I'll add a small notebook example and let you work on improving this. Ideally we'd want to work on a separate branch for this and not commit to master.

chiragnagpal avatar Oct 30 '20 00:10 chiragnagpal

I have added an Ipython notebook to the branch https://github.com/chiragnagpal/DeepSurvivalMachines/tree/usage_examples that trains and tests DSM in a 5 fold cross validation fashion and computes the Brier Score and Concordance Index:

https://github.com/chiragnagpal/DeepSurvivalMachines/blob/usage_examples/examples/SUPPORT_example.ipynb Next Steps:

  1. Add AUC(t) at the 25th, 50th and 75th event quantile.
  2. Compare to CoxPH, RSF and Weibull AFT
  3. Show how adding non linearities (Hidden layers of MLP) improves model perfromace.
  4. Bootstrap the results to generate confidence intervals around the metrics (This can be made a utility function in the library)
  5. Proper documentation with appropriate inline latex and markdown.

For hints on the kind of notebook I am talking about https://github.com/IBM/causallib/blob/master/examples/hemm_demo.ipynb

chiragnagpal avatar Oct 30 '20 01:10 chiragnagpal

@chiragnagpal Okay Sir... Understood.

kishanmaharaj avatar Oct 30 '20 04:10 kishanmaharaj

@chiragnagpal sir, can you send me some references on the above-mentioned models (DeepHit, DeepSurv, Random Survival Forest etc.)...?

kishanmaharaj avatar Oct 30 '20 05:10 kishanmaharaj

you can find the original references to the papers/methods mentioned in our paper https://arxiv.org/pdf/2003.01176.pdf see sec. 4.2

Fpor experiments, you might want to check out the python package pycox, scikit-survival and lifelines which have implementations of these methods.

chiragnagpal avatar Oct 30 '20 05:10 chiragnagpal

@chiragnagpal Okay, sir...

kishanmaharaj avatar Oct 30 '20 05:10 kishanmaharaj