graphs-pytorch

An implementation of basic graph layers in pytorch. Read the blog here.

vanillaGCN.py

A simplified Graph Convolutional Layer.

paperGCN.py

The Graph Convolutional Layer, as in the paper Semi-Supervised Classification with Graph Convolutional Networks.

multiHeadGraphAttention.py

A working and easy to understand implementation of multi-headed Graph Attention Layer.

Todo :

• Need to write tests for paperGCN and multiHeadedGraphAtt