X-CAL: Explicit Calibration for Survival Analysis

This repo contains the code for paper X-CAL: Explicit Calibration for Survival Analysis

This repo is derived from https://github.com/stanfordmlgroup/cdr-mimic

This repo works fine with python 3.7.4, pytorch 1.6.0, numpy 1.19.1, scipy 1.4.1 and lifeline 0.24.0.

train.sh will run an example of training the categorical model on the gamma dataset. Change the variable lam in the script will train the model with different levels of X-CAL regularization.

After train.sh finishes, test.sh will give the results on the test dataset.

If you use the code in your research, please cite the following publication

@article{goldstein2020x,
  title={X-CAL: Explicit Calibration for Survival Analysis},
  author={Goldstein, Mark and Han, Xintian and Puli, Aahlad and Perotte, Adler J and Ranganath, Rajesh},
  journal={Advances in neural information processing systems},
  volume={33},
  pages={18296},
  year={2020},
  publisher={NIH Public Access}
}