This code is for the following paper:
H. He, X. Yu, J. Zhang, S.H. Song, and K. B. Letaief, ‘‘Message passing meets graph neural networks: A new paradigm for massive MIMO systems,’’ IEEE Trans. Wireless Commun., DOI: 10.1109/TWC.2023.3321667, to appear. (https://arxiv.org/abs/2302.06896)
Please cite the paper if you use this code.
parsers.py: Simulation settings
main_with_genData.py: Run this function to train the network
mainTestScript.py: Run this function to test the trained model
Gen_dat.py: Generate the training and test data
GAMPNet.py: Network structure
Questions/suggestions/comments about AMP-GNN: eehthe@ust.hk / hehengtao137@163.com