Leveraging VAE for survival analysis
SAVAE: Survival Analysis Variational AutoEncoder
Developed a novel variational approach to enhance survival time predictions and manage complex, censored medical data.
Refined ELBO for survival analysis.
python survival_analysis/main_sota.py
If you want to display comparison results between SAVAE and SOTA models, run the following command:
python survival_analysis/results_display.py