DOCD

This is the source code for Dual Ontology-enhanced Clinical Decision Learning for First-admission Mortality Prediction

Data preparation

You will first need to request access for MIMIC dataset:

• MIMIC-III
• MIMIC-IV

Place the CSV files in the data directory

Model training

1. Install Required Libraries.

pip install -r requirements.txt

2. Process MIMIC data.

python DOCD_process_mimic3.py --out_path processed/
or
python DOCD_process_mimic4.py --out_path processed/

3. Train the model.

.\DOCD_mimic3.sh
or
.\DOCD_mimic4.sh

Cite

If you find the paper or the implementation helpful, please cite the following paper:

@article{yin2026dual,
  title={Dual Ontology-enhanced Clinical Decision Learning for First-admission Mortality Prediction},
  author={Yin, Fangchen and Nie, Hu and Pu, Xiaorong and Zheng, Pengcheng and Zhu, Qilin and Ren, Yazhou and Deng, Lijia},
  journal={IEEE Journal of Biomedical and Health Informatics},
  year={2026},
  publisher={IEEE}
}