/Calibrated-Survival-Analysis

Code Release for "Estimating Calibrated Individualized Survival Curves with Deep Learning" (Kamran & Wiens), AAAI 2021. https://www.aaai.org/AAAI21Papers/AAAI-8472.KamranF.pdf

Primary LanguagePython

Calibrated-Survival-Analysis

train.py sets up training for all potential methods of interest, with specific settings. This is the bulk of the training code.

util.py contains necessary utility functions for different training schemes.