This repository is for NIPS 2023 accepted paper: Ru, X., Zhang, X. Y., Moore, J. M., Liu, Z., & Yan, G. Attentive Transfer Entropy to Exploit Transient Emergence of Coupling Effect.
train_real.py -- the main program of our method;
load_data_real.py -- load the real coupled networks and data of activity sequences;
MI_estimator.py and attention_model.py -- the models of mutual information estimator for transfer entropy, and attention assigner in stage 1;
classifier.py -- the model of classifier in stage 2.