AI Features

Implementation of a GCN

Develop and train a Graph Convolutional Network (GCN) in Pytorch.

Implementing a 1-hop GCN layer in Pytorch

For this tutorial, we will train a simple 1-hop GCN layer in a small graph dataset. We will use the open-source graph data (you can find the link in the appendix) from the University of Dortmund. In particular, we will play with the MUTAG dataset because it is small enough to train something to solidify your understanding.

MUTAG dataset characteristics

Each node contains a label from 0 to 6 which will be used as a ...