KL-Regularized Entropic Optimal Transport for Adaptive Survival Analysis with Censored Data
- Classic Beran estimator implementation (
beran_baseline.py) - Proposed Iterative Beran + KL-regularized EOT algorithm (
beran_iterative_k_ti.py) - Survival function model with adaptive weighting (
survival_function_model.py) - Demo script with visualization and comparison (
demo.py) - Experiment saving and loading utilities (
save_results.py) - Fully reproducible
requirements.txt
- Clone the repository:
git clone https://github.com/NTAILab/Iterative_Beran_EOT.git cd Iterative_Beran_EOT