hehengtao / AMP_GNN

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: [e-mail] / [e-mail]