├── .gitignore ├── LICENSE ├── README.md ├── data └── kkbox_v1 │ ├── pycox_paper_figure4.png │ └── pycox_paper_table7.png ├── datasets └── kkbox_churn │ ├── basic_sessionization_ibis_duckdb_vs_polars.ipynb │ ├── csv_to_parquet.py │ ├── data_engineering.py │ ├── prepare_duckdb.py │ └── sessionization.py ├── demo_mlflow.py ├── model_selection ├── cross_validation.py └── wrappers.py ├── models ├── gradient_boosted_cif.py ├── kaplan_meier.py ├── kaplan_neighbors.py ├── kaplan_tree.py ├── meta_grid_bc.py ├── survival_mixin.py ├── tree_transformer.py └── yasgbt.py ├── notebooks ├── BrierScore.svg ├── censoring.png ├── kkbox_cv_benchmark.ipynb ├── kkbox_cv_yasgbt.ipynb ├── msk_mettropism.ipynb ├── truck_dataset.ipynb ├── truck_dataset.py ├── tutorial_part_1.ipynb ├── tutorial_part_1.py ├── tutorial_part_2.ipynb ├── tutorial_part_2.py └── variables.png └── plot ├── brier_score.py └── individuals.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/soda-inria/survival-analysis-benchmark/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/soda-inria/survival-analysis-benchmark/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/soda-inria/survival-analysis-benchmark/HEAD/README.md -------------------------------------------------------------------------------- /data/kkbox_v1/pycox_paper_figure4.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/soda-inria/survival-analysis-benchmark/HEAD/data/kkbox_v1/pycox_paper_figure4.png -------------------------------------------------------------------------------- /data/kkbox_v1/pycox_paper_table7.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/soda-inria/survival-analysis-benchmark/HEAD/data/kkbox_v1/pycox_paper_table7.png -------------------------------------------------------------------------------- /datasets/kkbox_churn/basic_sessionization_ibis_duckdb_vs_polars.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/soda-inria/survival-analysis-benchmark/HEAD/datasets/kkbox_churn/basic_sessionization_ibis_duckdb_vs_polars.ipynb -------------------------------------------------------------------------------- /datasets/kkbox_churn/csv_to_parquet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/soda-inria/survival-analysis-benchmark/HEAD/datasets/kkbox_churn/csv_to_parquet.py -------------------------------------------------------------------------------- /datasets/kkbox_churn/data_engineering.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/soda-inria/survival-analysis-benchmark/HEAD/datasets/kkbox_churn/data_engineering.py -------------------------------------------------------------------------------- /datasets/kkbox_churn/prepare_duckdb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/soda-inria/survival-analysis-benchmark/HEAD/datasets/kkbox_churn/prepare_duckdb.py -------------------------------------------------------------------------------- /datasets/kkbox_churn/sessionization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/soda-inria/survival-analysis-benchmark/HEAD/datasets/kkbox_churn/sessionization.py -------------------------------------------------------------------------------- /demo_mlflow.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/soda-inria/survival-analysis-benchmark/HEAD/demo_mlflow.py -------------------------------------------------------------------------------- /model_selection/cross_validation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/soda-inria/survival-analysis-benchmark/HEAD/model_selection/cross_validation.py -------------------------------------------------------------------------------- /model_selection/wrappers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/soda-inria/survival-analysis-benchmark/HEAD/model_selection/wrappers.py -------------------------------------------------------------------------------- /models/gradient_boosted_cif.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/soda-inria/survival-analysis-benchmark/HEAD/models/gradient_boosted_cif.py -------------------------------------------------------------------------------- /models/kaplan_meier.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/soda-inria/survival-analysis-benchmark/HEAD/models/kaplan_meier.py -------------------------------------------------------------------------------- /models/kaplan_neighbors.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/soda-inria/survival-analysis-benchmark/HEAD/models/kaplan_neighbors.py -------------------------------------------------------------------------------- /models/kaplan_tree.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/soda-inria/survival-analysis-benchmark/HEAD/models/kaplan_tree.py -------------------------------------------------------------------------------- /models/meta_grid_bc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/soda-inria/survival-analysis-benchmark/HEAD/models/meta_grid_bc.py -------------------------------------------------------------------------------- /models/survival_mixin.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/soda-inria/survival-analysis-benchmark/HEAD/models/survival_mixin.py -------------------------------------------------------------------------------- /models/tree_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/soda-inria/survival-analysis-benchmark/HEAD/models/tree_transformer.py -------------------------------------------------------------------------------- /models/yasgbt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/soda-inria/survival-analysis-benchmark/HEAD/models/yasgbt.py -------------------------------------------------------------------------------- /notebooks/BrierScore.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/soda-inria/survival-analysis-benchmark/HEAD/notebooks/BrierScore.svg -------------------------------------------------------------------------------- /notebooks/censoring.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/soda-inria/survival-analysis-benchmark/HEAD/notebooks/censoring.png -------------------------------------------------------------------------------- /notebooks/kkbox_cv_benchmark.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/soda-inria/survival-analysis-benchmark/HEAD/notebooks/kkbox_cv_benchmark.ipynb -------------------------------------------------------------------------------- /notebooks/kkbox_cv_yasgbt.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/soda-inria/survival-analysis-benchmark/HEAD/notebooks/kkbox_cv_yasgbt.ipynb -------------------------------------------------------------------------------- /notebooks/msk_mettropism.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/soda-inria/survival-analysis-benchmark/HEAD/notebooks/msk_mettropism.ipynb -------------------------------------------------------------------------------- /notebooks/truck_dataset.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/soda-inria/survival-analysis-benchmark/HEAD/notebooks/truck_dataset.ipynb -------------------------------------------------------------------------------- /notebooks/truck_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/soda-inria/survival-analysis-benchmark/HEAD/notebooks/truck_dataset.py -------------------------------------------------------------------------------- /notebooks/tutorial_part_1.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/soda-inria/survival-analysis-benchmark/HEAD/notebooks/tutorial_part_1.ipynb -------------------------------------------------------------------------------- /notebooks/tutorial_part_1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/soda-inria/survival-analysis-benchmark/HEAD/notebooks/tutorial_part_1.py -------------------------------------------------------------------------------- /notebooks/tutorial_part_2.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/soda-inria/survival-analysis-benchmark/HEAD/notebooks/tutorial_part_2.ipynb -------------------------------------------------------------------------------- /notebooks/tutorial_part_2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/soda-inria/survival-analysis-benchmark/HEAD/notebooks/tutorial_part_2.py -------------------------------------------------------------------------------- /notebooks/variables.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/soda-inria/survival-analysis-benchmark/HEAD/notebooks/variables.png -------------------------------------------------------------------------------- /plot/brier_score.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/soda-inria/survival-analysis-benchmark/HEAD/plot/brier_score.py -------------------------------------------------------------------------------- /plot/individuals.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/soda-inria/survival-analysis-benchmark/HEAD/plot/individuals.py --------------------------------------------------------------------------------