.gitignore
.pre-commit-config.yaml
LICENSE
README.md
conftest.py
pyproject.toml
.github/workflows/code-quality-main.yaml
.github/workflows/code-quality-pr.yaml
.github/workflows/python-build.yaml
.github/workflows/tests.yaml
src/MEDS_EIC_AR/__init__.py
src/MEDS_EIC_AR/__main__.py
src/MEDS_EIC_AR/utils.py
src/MEDS_EIC_AR.egg-info/PKG-INFO
src/MEDS_EIC_AR.egg-info/SOURCES.txt
src/MEDS_EIC_AR.egg-info/dependency_links.txt
src/MEDS_EIC_AR.egg-info/entry_points.txt
src/MEDS_EIC_AR.egg-info/requires.txt
src/MEDS_EIC_AR.egg-info/top_level.txt
src/MEDS_EIC_AR/configs/README.md
src/MEDS_EIC_AR/configs/__init__.py
src/MEDS_EIC_AR/configs/_demo_generate_trajectories.yaml
src/MEDS_EIC_AR/configs/_demo_pretrain.yaml
src/MEDS_EIC_AR/configs/_generate_trajectories.yaml
src/MEDS_EIC_AR/configs/_pretrain.yaml
src/MEDS_EIC_AR/configs/datamodule/__init__.py
src/MEDS_EIC_AR/configs/datamodule/default.yaml
src/MEDS_EIC_AR/configs/datamodule/generate_trajectories.yaml
src/MEDS_EIC_AR/configs/datamodule/pretrain.yaml
src/MEDS_EIC_AR/configs/inference/default.yaml
src/MEDS_EIC_AR/configs/inference/demo.yaml
src/MEDS_EIC_AR/configs/lightning_module/__init__.py
src/MEDS_EIC_AR/configs/lightning_module/default.yaml
src/MEDS_EIC_AR/configs/lightning_module/demo.yaml
src/MEDS_EIC_AR/configs/lightning_module/LR_scheduler/__init__.py
src/MEDS_EIC_AR/configs/lightning_module/LR_scheduler/cosine_annealing_warm_restarts.yaml
src/MEDS_EIC_AR/configs/lightning_module/LR_scheduler/get_cosine_schedule_with_warmup.yaml
src/MEDS_EIC_AR/configs/lightning_module/LR_scheduler/one_cycle_LR.yaml
src/MEDS_EIC_AR/configs/lightning_module/LR_scheduler/reduce_LR_on_plateau.yaml
src/MEDS_EIC_AR/configs/lightning_module/metrics/__init__.py
src/MEDS_EIC_AR/configs/lightning_module/metrics/default.yaml
src/MEDS_EIC_AR/configs/lightning_module/model/__init__.py
src/MEDS_EIC_AR/configs/lightning_module/model/default.yaml
src/MEDS_EIC_AR/configs/lightning_module/model/demo.yaml
src/MEDS_EIC_AR/configs/lightning_module/model/small.yaml
src/MEDS_EIC_AR/configs/lightning_module/optimizer/__init__.py
src/MEDS_EIC_AR/configs/lightning_module/optimizer/adam.yaml
src/MEDS_EIC_AR/configs/lightning_module/optimizer/adamw.yaml
src/MEDS_EIC_AR/configs/trainer/__init__.py
src/MEDS_EIC_AR/configs/trainer/default.yaml
src/MEDS_EIC_AR/configs/trainer/demo.yaml
src/MEDS_EIC_AR/configs/trainer/callbacks/__init__.py
src/MEDS_EIC_AR/configs/trainer/callbacks/default.yaml
src/MEDS_EIC_AR/configs/trainer/callbacks/early_stopping.yaml
src/MEDS_EIC_AR/configs/trainer/callbacks/learning_rate_monitor.yaml
src/MEDS_EIC_AR/configs/trainer/callbacks/model_checkpoint.yaml
src/MEDS_EIC_AR/configs/trainer/logger/__init__.py
src/MEDS_EIC_AR/configs/trainer/logger/csv.yaml
src/MEDS_EIC_AR/configs/trainer/logger/mlflow.yaml
src/MEDS_EIC_AR/configs/trainer/logger/wandb.yaml
src/MEDS_EIC_AR/generation/__init__.py
src/MEDS_EIC_AR/generation/format_trajectories.py
src/MEDS_EIC_AR/generation/utils.py
src/MEDS_EIC_AR/model/README.md
src/MEDS_EIC_AR/model/__init__.py
src/MEDS_EIC_AR/model/model.py
src/MEDS_EIC_AR/preprocessing/README.md
src/MEDS_EIC_AR/preprocessing/__init__.py
src/MEDS_EIC_AR/preprocessing/__main__.py
src/MEDS_EIC_AR/preprocessing/configs/_data.yaml
src/MEDS_EIC_AR/preprocessing/configs/_process_data.yaml
src/MEDS_EIC_AR/preprocessing/configs/_reshard_data.yaml
src/MEDS_EIC_AR/training/README.md
src/MEDS_EIC_AR/training/__init__.py
src/MEDS_EIC_AR/training/metrics.py
src/MEDS_EIC_AR/training/module.py
tests/__init__.py
tests/test_generate_trajectories.py
tests/test_pretrain.py
tests/test_process_data.py