Power Allocation for Wireless Federated Learning using Graph Neural Networks

This code package is related to the following scientific article:

Boning Li, Ananthram Swami, and Santiago Segarra, "Power Allocation for Wireless Federated Learning using Graph Neural Networks,"