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.
(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.