/Graph-Neural-Networks-INF367A

Traffic prediction with graph neural network using PyTorch Geometric. The implementation uses the MetaLayer class to build the GNN which allows for separate edge, node and global models.

Primary LanguagePython

Stargazers