Torch Geometric GNN

An implementation of a graph neural network using the torch_geometric library

Example output

Enter the number of nodes: 5
Predicted sequence: [3. 1. 4. 2. 0.] | Target sequence: [3. 1. 4. 2. 0.]