Wileless-PDGNet Implementation
This repo is related to the following paper:
Boning Li, Ananthram Swami, and Santiago Segarra, "Power allocation for wireless federated learning using graph neural networks," submitted to ICASSP 2022.
Data Generation
Please refer to https://github.com/bmatthiesen/deep-EE-opt/tree/master/data to generate channel simulations.
Power Allocation
See `main.py'.
Federated Learning
See `fl-main.ipynb'