.gitignore
.pre-commit-config.yaml
.project-root
.readthedocs.yaml
LICENSE
Makefile
README.md
mkdocs.yml
pyproject.toml
.github/workflows/code-quality-main.yaml
.github/workflows/code-quality-pr.yaml
.github/workflows/publish-to-pypi.yml
.github/workflows/tests.yaml
MIMICIV_TUTORIAL/README.MD
MIMICIV_TUTORIAL/generate_ebcl_windows.py
MIMICIV_TUTORIAL/post_meds_extract.sh
data/.gitkeep
docs/gen_ref_pages.py
docs/index.md
docs/input_encoding_methods.md
docs/terminology.md
docs/tutorial.md
docs/assets/light_logo.png
docs/javascripts/mathjax.js
docs/overrides/main.html
notebooks/.gitkeep
scripts/schedule.sh
src/meds_torch/__init__.py
src/meds_torch/eval.py
src/meds_torch/train.py
src/meds_torch/transfer_learning.py
src/meds_torch/tune.py
src/meds_torch.egg-info/PKG-INFO
src/meds_torch.egg-info/SOURCES.txt
src/meds_torch.egg-info/dependency_links.txt
src/meds_torch.egg-info/entry_points.txt
src/meds_torch.egg-info/requires.txt
src/meds_torch.egg-info/top_level.txt
src/meds_torch/configs/__init__.py
src/meds_torch/configs/eval.yaml
src/meds_torch/configs/train.yaml
src/meds_torch/configs/transfer_learning.yaml
src/meds_torch/configs/callbacks/__init__.py
src/meds_torch/configs/callbacks/default.yaml
src/meds_torch/configs/callbacks/early_stopping.yaml
src/meds_torch/configs/callbacks/model_checkpoint.yaml
src/meds_torch/configs/callbacks/model_summary.yaml
src/meds_torch/configs/callbacks/none.yaml
src/meds_torch/configs/callbacks/ray.yaml
src/meds_torch/configs/callbacks/rich_progress_bar.yaml
src/meds_torch/configs/data/__init__.py
src/meds_torch/configs/data/multiwindow_pytorch_dataset.yaml
src/meds_torch/configs/data/pytorch_dataset.yaml
src/meds_torch/configs/debug/__init__.py
src/meds_torch/configs/debug/default.yaml
src/meds_torch/configs/debug/fdr.yaml
src/meds_torch/configs/debug/limit.yaml
src/meds_torch/configs/debug/overfit.yaml
src/meds_torch/configs/debug/profiler.yaml
src/meds_torch/configs/experiment/__init__.py
src/meds_torch/configs/experiment/example.yaml
src/meds_torch/configs/experiment/mimiciv.yaml
src/meds_torch/configs/extras/__init__.py
src/meds_torch/configs/extras/default.yaml
src/meds_torch/configs/hparams_search/__init__.py
src/meds_torch/configs/hparams_search/optuna.yaml
src/meds_torch/configs/hparams_search/ray_tune.yaml
src/meds_torch/configs/hydra/__init__.py
src/meds_torch/configs/hydra/default.yaml
src/meds_torch/configs/logger/__init__.py
src/meds_torch/configs/logger/aim.yaml
src/meds_torch/configs/logger/comet.yaml
src/meds_torch/configs/logger/csv.yaml
src/meds_torch/configs/logger/many_loggers.yaml
src/meds_torch/configs/logger/mlflow.yaml
src/meds_torch/configs/logger/neptune.yaml
src/meds_torch/configs/logger/tensorboard.yaml
src/meds_torch/configs/logger/wandb.yaml
src/meds_torch/configs/model/__init__.py
src/meds_torch/configs/model/ebcl.yaml
src/meds_torch/configs/model/ocp.yaml
src/meds_torch/configs/model/supervised.yaml
src/meds_torch/configs/model/token_forecasting.yaml
src/meds_torch/configs/model/value_forecasting.yaml
src/meds_torch/configs/model/backbone/__init__.py
src/meds_torch/configs/model/backbone/default.yaml
src/meds_torch/configs/model/backbone/lstm.yaml
src/meds_torch/configs/model/backbone/mamba.yaml
src/meds_torch/configs/model/backbone/transformer_decoder.yaml
src/meds_torch/configs/model/backbone/transformer_encoder.yaml
src/meds_torch/configs/model/backbone/transformer_encoder_attn_avg.yaml
src/meds_torch/configs/model/input_encoder/__init__.py
src/meds_torch/configs/model/input_encoder/all_text_encoder.yaml
src/meds_torch/configs/model/input_encoder/only_code_encoder.yaml
src/meds_torch/configs/model/input_encoder/text_code_encoder.yaml
src/meds_torch/configs/model/input_encoder/text_observation_encoder.yaml
src/meds_torch/configs/model/input_encoder/triplet_encoder.yaml
src/meds_torch/configs/model/input_encoder/triplet_prompt_encoder.yaml
src/meds_torch/configs/paths/__init__.py
src/meds_torch/configs/paths/default.yaml
src/meds_torch/configs/trainer/__init__.py
src/meds_torch/configs/trainer/cpu.yaml
src/meds_torch/configs/trainer/ddp.yaml
src/meds_torch/configs/trainer/ddp_sim.yaml
src/meds_torch/configs/trainer/default.yaml
src/meds_torch/configs/trainer/gpu.yaml
src/meds_torch/configs/trainer/mps.yaml
src/meds_torch/configs/trainer/ray.yaml
src/meds_torch/data/__init__.py
src/meds_torch/data/datamodule.py
src/meds_torch/data/components/__init__.py
src/meds_torch/data/components/multiwindow_pytorch_dataset.py
src/meds_torch/data/components/pytorch_dataset.py
src/meds_torch/input_encoder/__init__.py
src/meds_torch/input_encoder/text_encoder.py
src/meds_torch/input_encoder/triplet_encoder.py
src/meds_torch/input_encoder/triplet_prompt_encoder.py
src/meds_torch/models/__init__.py
src/meds_torch/models/base_model.py
src/meds_torch/models/ebcl_model.py
src/meds_torch/models/ocp_model.py
src/meds_torch/models/supervised_model.py
src/meds_torch/models/token_forecasting.py
src/meds_torch/models/utils.py
src/meds_torch/models/value_forecasting.py
src/meds_torch/models/components/__init__.py
src/meds_torch/models/components/lstm.py
src/meds_torch/models/components/mamba.py
src/meds_torch/models/components/transformer_decoder.py
src/meds_torch/models/components/transformer_encoder.py
src/meds_torch/models/components/transformer_encoder_attn_avg.py
src/meds_torch/models/components/utils.py
src/meds_torch/utils/__init__.py
src/meds_torch/utils/custom_normalization.py
src/meds_torch/utils/custom_time_token.py
src/meds_torch/utils/instantiators.py
src/meds_torch/utils/logging_utils.py
src/meds_torch/utils/module_class.py
src/meds_torch/utils/pylogger.py
src/meds_torch/utils/resolvers.py
src/meds_torch/utils/rich_utils.py
src/meds_torch/utils/utils.py
tests/__init__.py
tests/conftest.py
tests/test_cli.py
tests/test_configs.py
tests/test_datamodules.py
tests/test_eval.py
tests/test_train.py
tests/test_transfer_learning.py
tests/helpers/__init__.py
tests/helpers/extract_test_data.py
tests/helpers/generate_test_data.sh
tests/helpers/generate_test_data_tensors.py
tests/helpers/generate_test_windows.py
tests/helpers/package_available.py
tests/helpers/run_if.py
tests/helpers/run_sh_command.py
tests/helpers/triplet_tensorization.yaml
tests/helpers/utils.py
tests/test_data/MEDS_cohort/convert_to_sharded_events/event_conversion_config.yaml
tests/test_data/MEDS_cohort/convert_to_sharded_events/held_out/0/admit_vitals/[0-100).parquet
tests/test_data/MEDS_cohort/convert_to_sharded_events/held_out/0/admit_vitals/[100-200).parquet
tests/test_data/MEDS_cohort/convert_to_sharded_events/held_out/0/admit_vitals/[200-300).parquet
tests/test_data/MEDS_cohort/convert_to_sharded_events/held_out/0/admit_vitals/[300-400).parquet
tests/test_data/MEDS_cohort/convert_to_sharded_events/held_out/0/admit_vitals/[400-500).parquet
tests/test_data/MEDS_cohort/convert_to_sharded_events/held_out/0/admit_vitals/[500-578).parquet
tests/test_data/MEDS_cohort/convert_to_sharded_events/held_out/0/subjects/[0-100).parquet
tests/test_data/MEDS_cohort/convert_to_sharded_events/train/0/admit_vitals/[0-100).parquet
tests/test_data/MEDS_cohort/convert_to_sharded_events/train/0/admit_vitals/[100-200).parquet
tests/test_data/MEDS_cohort/convert_to_sharded_events/train/0/admit_vitals/[200-300).parquet
tests/test_data/MEDS_cohort/convert_to_sharded_events/train/0/admit_vitals/[300-400).parquet
tests/test_data/MEDS_cohort/convert_to_sharded_events/train/0/admit_vitals/[400-500).parquet
tests/test_data/MEDS_cohort/convert_to_sharded_events/train/0/admit_vitals/[500-578).parquet
tests/test_data/MEDS_cohort/convert_to_sharded_events/train/0/subjects/[0-100).parquet
tests/test_data/MEDS_cohort/convert_to_sharded_events/train/1/admit_vitals/[0-100).parquet
tests/test_data/MEDS_cohort/convert_to_sharded_events/train/1/admit_vitals/[100-200).parquet
tests/test_data/MEDS_cohort/convert_to_sharded_events/train/1/admit_vitals/[200-300).parquet
tests/test_data/MEDS_cohort/convert_to_sharded_events/train/1/admit_vitals/[300-400).parquet
tests/test_data/MEDS_cohort/convert_to_sharded_events/train/1/admit_vitals/[400-500).parquet
tests/test_data/MEDS_cohort/convert_to_sharded_events/train/1/admit_vitals/[500-578).parquet
tests/test_data/MEDS_cohort/convert_to_sharded_events/train/1/subjects/[0-100).parquet
tests/test_data/MEDS_cohort/convert_to_sharded_events/train/2/admit_vitals/[0-100).parquet
tests/test_data/MEDS_cohort/convert_to_sharded_events/train/2/admit_vitals/[100-200).parquet
tests/test_data/MEDS_cohort/convert_to_sharded_events/train/2/admit_vitals/[200-300).parquet
tests/test_data/MEDS_cohort/convert_to_sharded_events/train/2/admit_vitals/[300-400).parquet
tests/test_data/MEDS_cohort/convert_to_sharded_events/train/2/admit_vitals/[400-500).parquet
tests/test_data/MEDS_cohort/convert_to_sharded_events/train/2/admit_vitals/[500-578).parquet
tests/test_data/MEDS_cohort/convert_to_sharded_events/train/2/subjects/[0-100).parquet
tests/test_data/MEDS_cohort/convert_to_sharded_events/train/3/admit_vitals/[0-100).parquet
tests/test_data/MEDS_cohort/convert_to_sharded_events/train/3/admit_vitals/[100-200).parquet
tests/test_data/MEDS_cohort/convert_to_sharded_events/train/3/admit_vitals/[200-300).parquet
tests/test_data/MEDS_cohort/convert_to_sharded_events/train/3/admit_vitals/[300-400).parquet
tests/test_data/MEDS_cohort/convert_to_sharded_events/train/3/admit_vitals/[400-500).parquet
tests/test_data/MEDS_cohort/convert_to_sharded_events/train/3/admit_vitals/[500-578).parquet
tests/test_data/MEDS_cohort/convert_to_sharded_events/train/3/subjects/[0-100).parquet
tests/test_data/MEDS_cohort/convert_to_sharded_events/tuning/0/admit_vitals/[0-100).parquet
tests/test_data/MEDS_cohort/convert_to_sharded_events/tuning/0/admit_vitals/[100-200).parquet
tests/test_data/MEDS_cohort/convert_to_sharded_events/tuning/0/admit_vitals/[200-300).parquet
tests/test_data/MEDS_cohort/convert_to_sharded_events/tuning/0/admit_vitals/[300-400).parquet
tests/test_data/MEDS_cohort/convert_to_sharded_events/tuning/0/admit_vitals/[400-500).parquet
tests/test_data/MEDS_cohort/convert_to_sharded_events/tuning/0/admit_vitals/[500-578).parquet
tests/test_data/MEDS_cohort/convert_to_sharded_events/tuning/0/subjects/[0-100).parquet
tests/test_data/MEDS_cohort/data/held_out/0.parquet
tests/test_data/MEDS_cohort/data/train/0.parquet
tests/test_data/MEDS_cohort/data/train/1.parquet
tests/test_data/MEDS_cohort/data/train/2.parquet
tests/test_data/MEDS_cohort/data/train/3.parquet
tests/test_data/MEDS_cohort/data/tuning/0.parquet
tests/test_data/MEDS_cohort/extract_code_metadata/codes.parquet
tests/test_data/MEDS_cohort/extract_code_metadata/demo_metadata.parquet
tests/test_data/MEDS_cohort/extract_code_metadata/event_conversion_config.yaml
tests/test_data/MEDS_cohort/extract_code_metadata/input_metadata.parquet
tests/test_data/MEDS_cohort/merge_to_MEDS_cohort/held_out/0.parquet
tests/test_data/MEDS_cohort/merge_to_MEDS_cohort/train/0.parquet
tests/test_data/MEDS_cohort/merge_to_MEDS_cohort/train/1.parquet
tests/test_data/MEDS_cohort/merge_to_MEDS_cohort/train/2.parquet
tests/test_data/MEDS_cohort/merge_to_MEDS_cohort/train/3.parquet
tests/test_data/MEDS_cohort/merge_to_MEDS_cohort/tuning/0.parquet
tests/test_data/MEDS_cohort/metadata/.shards.json
tests/test_data/MEDS_cohort/metadata/codes.parquet
tests/test_data/MEDS_cohort/metadata/dataset.json
tests/test_data/MEDS_cohort/metadata/patient_splits.parquet
tests/test_data/MEDS_cohort/shard_events/admit_vitals/[0-100).parquet
tests/test_data/MEDS_cohort/shard_events/admit_vitals/[100-200).parquet
tests/test_data/MEDS_cohort/shard_events/admit_vitals/[200-300).parquet
tests/test_data/MEDS_cohort/shard_events/admit_vitals/[300-400).parquet
tests/test_data/MEDS_cohort/shard_events/admit_vitals/[400-500).parquet
tests/test_data/MEDS_cohort/shard_events/admit_vitals/[500-578).parquet
tests/test_data/MEDS_cohort/shard_events/subjects/[0-100).parquet
tests/test_data/eic_tensors/aggregate_code_metadata/codes.parquet
tests/test_data/eic_tensors/aggregate_code_metadata/train/0.parquet
tests/test_data/eic_tensors/aggregate_code_metadata/train/1.parquet
tests/test_data/eic_tensors/aggregate_code_metadata/train/2.parquet
tests/test_data/eic_tensors/aggregate_code_metadata/train/3.parquet
tests/test_data/eic_tensors/custom_normalization/held_out/0.parquet
tests/test_data/eic_tensors/custom_normalization/train/0.parquet
tests/test_data/eic_tensors/custom_normalization/train/1.parquet
tests/test_data/eic_tensors/custom_normalization/train/2.parquet
tests/test_data/eic_tensors/custom_normalization/train/3.parquet
tests/test_data/eic_tensors/custom_normalization/tuning/0.parquet
tests/test_data/eic_tensors/custom_time_token/held_out/0.parquet
tests/test_data/eic_tensors/custom_time_token/train/0.parquet
tests/test_data/eic_tensors/custom_time_token/train/1.parquet
tests/test_data/eic_tensors/custom_time_token/train/2.parquet
tests/test_data/eic_tensors/custom_time_token/train/3.parquet
tests/test_data/eic_tensors/custom_time_token/tuning/0.parquet
tests/test_data/eic_tensors/data/held_out/0.nrt
tests/test_data/eic_tensors/data/train/0.nrt
tests/test_data/eic_tensors/data/train/1.nrt
tests/test_data/eic_tensors/data/train/2.nrt
tests/test_data/eic_tensors/data/train/3.nrt
tests/test_data/eic_tensors/data/tuning/0.nrt
tests/test_data/eic_tensors/filter_measurements/held_out/0.parquet
tests/test_data/eic_tensors/filter_measurements/train/0.parquet
tests/test_data/eic_tensors/filter_measurements/train/1.parquet
tests/test_data/eic_tensors/filter_measurements/train/2.parquet
tests/test_data/eic_tensors/filter_measurements/train/3.parquet
tests/test_data/eic_tensors/filter_measurements/tuning/0.parquet
tests/test_data/eic_tensors/filter_patients/held_out/0.parquet
tests/test_data/eic_tensors/filter_patients/train/0.parquet
tests/test_data/eic_tensors/filter_patients/train/1.parquet
tests/test_data/eic_tensors/filter_patients/train/2.parquet
tests/test_data/eic_tensors/filter_patients/train/3.parquet
tests/test_data/eic_tensors/filter_patients/tuning/0.parquet
tests/test_data/eic_tensors/metadata/codes.parquet
tests/test_data/eic_tensors/tokenization/event_seqs/held_out/0.parquet
tests/test_data/eic_tensors/tokenization/event_seqs/train/0.parquet
tests/test_data/eic_tensors/tokenization/event_seqs/train/1.parquet
tests/test_data/eic_tensors/tokenization/event_seqs/train/2.parquet
tests/test_data/eic_tensors/tokenization/event_seqs/train/3.parquet
tests/test_data/eic_tensors/tokenization/event_seqs/tuning/0.parquet
tests/test_data/eic_tensors/tokenization/schemas/held_out/0.parquet
tests/test_data/eic_tensors/tokenization/schemas/train/0.parquet
tests/test_data/eic_tensors/tokenization/schemas/train/1.parquet
tests/test_data/eic_tensors/tokenization/schemas/train/2.parquet
tests/test_data/eic_tensors/tokenization/schemas/train/3.parquet
tests/test_data/eic_tensors/tokenization/schemas/tuning/0.parquet
tests/test_data/raw_cohort/admit_vitals.csv
tests/test_data/raw_cohort/admit_vitals.parquet
tests/test_data/raw_cohort/demo_metadata.csv
tests/test_data/raw_cohort/event_cfgs.yaml
tests/test_data/raw_cohort/input_metadata.csv
tests/test_data/raw_cohort/subjects.csv
tests/test_data/triplet_tensors/add_time_derived_measurements/held_out/0.parquet
tests/test_data/triplet_tensors/add_time_derived_measurements/train/0.parquet
tests/test_data/triplet_tensors/add_time_derived_measurements/train/1.parquet
tests/test_data/triplet_tensors/add_time_derived_measurements/train/2.parquet
tests/test_data/triplet_tensors/add_time_derived_measurements/train/3.parquet
tests/test_data/triplet_tensors/add_time_derived_measurements/tuning/0.parquet
tests/test_data/triplet_tensors/aggregate_code_metadata/codes.parquet
tests/test_data/triplet_tensors/aggregate_code_metadata/train/0.parquet
tests/test_data/triplet_tensors/aggregate_code_metadata/train/1.parquet
tests/test_data/triplet_tensors/aggregate_code_metadata/train/2.parquet
tests/test_data/triplet_tensors/aggregate_code_metadata/train/3.parquet
tests/test_data/triplet_tensors/data/held_out/0.nrt
tests/test_data/triplet_tensors/data/train/0.nrt
tests/test_data/triplet_tensors/data/train/1.nrt
tests/test_data/triplet_tensors/data/train/2.nrt
tests/test_data/triplet_tensors/data/train/3.nrt
tests/test_data/triplet_tensors/data/tuning/0.nrt
tests/test_data/triplet_tensors/filter_measurements/held_out/0.parquet
tests/test_data/triplet_tensors/filter_measurements/train/0.parquet
tests/test_data/triplet_tensors/filter_measurements/train/1.parquet
tests/test_data/triplet_tensors/filter_measurements/train/2.parquet
tests/test_data/triplet_tensors/filter_measurements/train/3.parquet
tests/test_data/triplet_tensors/filter_measurements/tuning/0.parquet
tests/test_data/triplet_tensors/filter_patients/held_out/0.parquet
tests/test_data/triplet_tensors/filter_patients/train/0.parquet
tests/test_data/triplet_tensors/filter_patients/train/1.parquet
tests/test_data/triplet_tensors/filter_patients/train/2.parquet
tests/test_data/triplet_tensors/filter_patients/train/3.parquet
tests/test_data/triplet_tensors/filter_patients/tuning/0.parquet
tests/test_data/triplet_tensors/metadata/codes.parquet
tests/test_data/triplet_tensors/normalization/held_out/0.parquet
tests/test_data/triplet_tensors/normalization/train/0.parquet
tests/test_data/triplet_tensors/normalization/train/1.parquet
tests/test_data/triplet_tensors/normalization/train/2.parquet
tests/test_data/triplet_tensors/normalization/train/3.parquet
tests/test_data/triplet_tensors/normalization/tuning/0.parquet
tests/test_data/triplet_tensors/occlude_outliers/held_out/0.parquet
tests/test_data/triplet_tensors/occlude_outliers/train/0.parquet
tests/test_data/triplet_tensors/occlude_outliers/train/1.parquet
tests/test_data/triplet_tensors/occlude_outliers/train/2.parquet
tests/test_data/triplet_tensors/occlude_outliers/train/3.parquet
tests/test_data/triplet_tensors/occlude_outliers/tuning/0.parquet
tests/test_data/triplet_tensors/tokenization/event_seqs/held_out/0.parquet
tests/test_data/triplet_tensors/tokenization/event_seqs/train/0.parquet
tests/test_data/triplet_tensors/tokenization/event_seqs/train/1.parquet
tests/test_data/triplet_tensors/tokenization/event_seqs/train/2.parquet
tests/test_data/triplet_tensors/tokenization/event_seqs/train/3.parquet
tests/test_data/triplet_tensors/tokenization/event_seqs/tuning/0.parquet
tests/test_data/triplet_tensors/tokenization/schemas/held_out/0.parquet
tests/test_data/triplet_tensors/tokenization/schemas/train/0.parquet
tests/test_data/triplet_tensors/tokenization/schemas/train/1.parquet
tests/test_data/triplet_tensors/tokenization/schemas/train/2.parquet
tests/test_data/triplet_tensors/tokenization/schemas/train/3.parquet
tests/test_data/triplet_tensors/tokenization/schemas/tuning/0.parquet
tests/test_data/windows/raw_windows.parquet