Implementation of our RAG-IM (Retrieval Augmented Generation of Interpretable Models) paper.
CAUTION: The codebase is being actively modified.
Please download the MIMIC-IV dataset from the official source. The following files are required:
- HOSP.procedures_icd.csv
- HOSP.d_icd_procedures.csv
- HOSP.d_icd_diagnoses.csv
- HOSP.d_labitems.csv
- ED.diagnosis.csv
- labevents.csv
bash run_experiment.sh
bash run_test_only.sh