This repository is for AAAI-23 accepted paper: https://ojs.aaai.org/index.php/AAAI/article/view/26152
Ru, X., Moore, J. M., Zhang, X. Y., Zeng, Y., & Yan, G. (2023, June). Inferring patient zero on temporal networks via graph neural networks. In Proceedings of the AAAI Conference on Artificial Intelligence (Vol. 37, No. 8, pp. 9632-9640).
gnn_model.py(model) + load_main.py (data )+ train_main.py ( training )
gnn_model.py (model) + load_linkmiss.py (data )+ train_linkmiss.py ( training )
gnn_model.py (model) + load_unobserved.py (data )+ train_unobserved.py ( training )
data is generated by SIR.py