├── LICENSE ├── README.md ├── T5 ├── abstain_with_entropy.py ├── config.py ├── config │ └── ehrsql │ │ ├── eval │ │ ├── ehrsql_eicu_t5_base__eicu_valid.yaml │ │ ├── ehrsql_eicu_t5_base_schema__eicu_valid.yaml │ │ ├── ehrsql_mimic3_t5_base__mimic3_valid.yaml │ │ └── ehrsql_mimic3_t5_base_schema__mimic3_valid.yaml │ │ └── training │ │ ├── ehrsql_eicu_t5_base.yaml │ │ ├── ehrsql_eicu_t5_base_schema.yaml │ │ ├── ehrsql_mimic3_t5_base.yaml │ │ └── ehrsql_mimic3_t5_base_schema.yaml ├── generate.py ├── main.py ├── model.py └── trainer_t5.py ├── dataset └── ehrsql │ ├── eicu │ ├── eicu.sql │ ├── test.json │ ├── train.json │ └── valid.json │ ├── mimic_iii │ ├── mimic_iii.sql │ ├── test.json │ ├── train.json │ └── valid.json │ └── tables.json ├── evaluate.py ├── outputs ├── eval_ehrsql_mimic3_t5_base__mimic3_valid │ ├── prediction.json │ └── prediction_raw.json └── eval_ehrsql_mimic3_t5_base_schema__mimic3_valid │ └── prediction_raw.json ├── preprocess ├── preprocess_db.py ├── preprocess_db_eicu.py ├── preprocess_db_mimic_iii.py └── preprocess_utils.py ├── t5_threshold.ipynb └── utils ├── dataset.py ├── logger.py ├── model_utils.py └── optim.py /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/glee4810/EHRSQL/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/glee4810/EHRSQL/HEAD/README.md -------------------------------------------------------------------------------- /T5/abstain_with_entropy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/glee4810/EHRSQL/HEAD/T5/abstain_with_entropy.py -------------------------------------------------------------------------------- /T5/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/glee4810/EHRSQL/HEAD/T5/config.py -------------------------------------------------------------------------------- /T5/config/ehrsql/eval/ehrsql_eicu_t5_base__eicu_valid.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/glee4810/EHRSQL/HEAD/T5/config/ehrsql/eval/ehrsql_eicu_t5_base__eicu_valid.yaml -------------------------------------------------------------------------------- /T5/config/ehrsql/eval/ehrsql_eicu_t5_base_schema__eicu_valid.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/glee4810/EHRSQL/HEAD/T5/config/ehrsql/eval/ehrsql_eicu_t5_base_schema__eicu_valid.yaml -------------------------------------------------------------------------------- /T5/config/ehrsql/eval/ehrsql_mimic3_t5_base__mimic3_valid.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/glee4810/EHRSQL/HEAD/T5/config/ehrsql/eval/ehrsql_mimic3_t5_base__mimic3_valid.yaml -------------------------------------------------------------------------------- /T5/config/ehrsql/eval/ehrsql_mimic3_t5_base_schema__mimic3_valid.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/glee4810/EHRSQL/HEAD/T5/config/ehrsql/eval/ehrsql_mimic3_t5_base_schema__mimic3_valid.yaml -------------------------------------------------------------------------------- /T5/config/ehrsql/training/ehrsql_eicu_t5_base.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/glee4810/EHRSQL/HEAD/T5/config/ehrsql/training/ehrsql_eicu_t5_base.yaml -------------------------------------------------------------------------------- /T5/config/ehrsql/training/ehrsql_eicu_t5_base_schema.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/glee4810/EHRSQL/HEAD/T5/config/ehrsql/training/ehrsql_eicu_t5_base_schema.yaml -------------------------------------------------------------------------------- /T5/config/ehrsql/training/ehrsql_mimic3_t5_base.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/glee4810/EHRSQL/HEAD/T5/config/ehrsql/training/ehrsql_mimic3_t5_base.yaml -------------------------------------------------------------------------------- /T5/config/ehrsql/training/ehrsql_mimic3_t5_base_schema.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/glee4810/EHRSQL/HEAD/T5/config/ehrsql/training/ehrsql_mimic3_t5_base_schema.yaml -------------------------------------------------------------------------------- /T5/generate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/glee4810/EHRSQL/HEAD/T5/generate.py -------------------------------------------------------------------------------- /T5/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/glee4810/EHRSQL/HEAD/T5/main.py -------------------------------------------------------------------------------- /T5/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/glee4810/EHRSQL/HEAD/T5/model.py -------------------------------------------------------------------------------- /T5/trainer_t5.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/glee4810/EHRSQL/HEAD/T5/trainer_t5.py -------------------------------------------------------------------------------- /dataset/ehrsql/eicu/eicu.sql: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/glee4810/EHRSQL/HEAD/dataset/ehrsql/eicu/eicu.sql -------------------------------------------------------------------------------- /dataset/ehrsql/eicu/test.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/glee4810/EHRSQL/HEAD/dataset/ehrsql/eicu/test.json -------------------------------------------------------------------------------- /dataset/ehrsql/eicu/train.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/glee4810/EHRSQL/HEAD/dataset/ehrsql/eicu/train.json -------------------------------------------------------------------------------- /dataset/ehrsql/eicu/valid.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/glee4810/EHRSQL/HEAD/dataset/ehrsql/eicu/valid.json -------------------------------------------------------------------------------- /dataset/ehrsql/mimic_iii/mimic_iii.sql: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/glee4810/EHRSQL/HEAD/dataset/ehrsql/mimic_iii/mimic_iii.sql -------------------------------------------------------------------------------- /dataset/ehrsql/mimic_iii/test.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/glee4810/EHRSQL/HEAD/dataset/ehrsql/mimic_iii/test.json -------------------------------------------------------------------------------- /dataset/ehrsql/mimic_iii/train.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/glee4810/EHRSQL/HEAD/dataset/ehrsql/mimic_iii/train.json -------------------------------------------------------------------------------- /dataset/ehrsql/mimic_iii/valid.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/glee4810/EHRSQL/HEAD/dataset/ehrsql/mimic_iii/valid.json -------------------------------------------------------------------------------- /dataset/ehrsql/tables.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/glee4810/EHRSQL/HEAD/dataset/ehrsql/tables.json -------------------------------------------------------------------------------- /evaluate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/glee4810/EHRSQL/HEAD/evaluate.py -------------------------------------------------------------------------------- /outputs/eval_ehrsql_mimic3_t5_base__mimic3_valid/prediction.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/glee4810/EHRSQL/HEAD/outputs/eval_ehrsql_mimic3_t5_base__mimic3_valid/prediction.json -------------------------------------------------------------------------------- /outputs/eval_ehrsql_mimic3_t5_base__mimic3_valid/prediction_raw.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/glee4810/EHRSQL/HEAD/outputs/eval_ehrsql_mimic3_t5_base__mimic3_valid/prediction_raw.json -------------------------------------------------------------------------------- /outputs/eval_ehrsql_mimic3_t5_base_schema__mimic3_valid/prediction_raw.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/glee4810/EHRSQL/HEAD/outputs/eval_ehrsql_mimic3_t5_base_schema__mimic3_valid/prediction_raw.json -------------------------------------------------------------------------------- /preprocess/preprocess_db.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/glee4810/EHRSQL/HEAD/preprocess/preprocess_db.py -------------------------------------------------------------------------------- /preprocess/preprocess_db_eicu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/glee4810/EHRSQL/HEAD/preprocess/preprocess_db_eicu.py -------------------------------------------------------------------------------- /preprocess/preprocess_db_mimic_iii.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/glee4810/EHRSQL/HEAD/preprocess/preprocess_db_mimic_iii.py -------------------------------------------------------------------------------- /preprocess/preprocess_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/glee4810/EHRSQL/HEAD/preprocess/preprocess_utils.py -------------------------------------------------------------------------------- /t5_threshold.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/glee4810/EHRSQL/HEAD/t5_threshold.ipynb -------------------------------------------------------------------------------- /utils/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/glee4810/EHRSQL/HEAD/utils/dataset.py -------------------------------------------------------------------------------- /utils/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/glee4810/EHRSQL/HEAD/utils/logger.py -------------------------------------------------------------------------------- /utils/model_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/glee4810/EHRSQL/HEAD/utils/model_utils.py -------------------------------------------------------------------------------- /utils/optim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/glee4810/EHRSQL/HEAD/utils/optim.py --------------------------------------------------------------------------------