/TAMSGC2021

The code of the paper "Medication Combination Prediction Using Temporal Attention Mechanism and Simple Graph Convolution" of IEEE Journal of Biomedical and Health Informatics

Primary LanguagePython

TAMSGC2021

The code related to the paper below:

Haiqiang Wang, Yinying Wu, Chao Gao*, Yue Deng, Fan Zhang, Jiajin Huang, Jiming Liu*, Medication Combination Prediction Using Temporal Attention Mechanism and Simple Graph Convolution, IEEE Journal of Biomedical and Health Informatics, 2021, doi: 10.1109/JBHI.2021.3082548.

Running the code

Data preprocessing

(1) In ./data, you can find the well-preprocessed sample data in pickle form. These sample data are just a small part of the complete data, you can use them to test these code. The complete data can be downloaded from https://mimic.physionet.org/

(2) In ./data, you can find our main code, the train_model.py and the test_model.py are used to train and test our model, respectively.

(3) In ./saved, you can find the saved model of each training epoch.