#4·gcn

Graph Classification & Batchwise Training

Author: rushilanirudhCreated Jan 19, 2017Updated Feb 13, 2024

Hi I have two questions:

  • I would like to repurpose this for graph classification (where I have a single label per graph). I see there is an option to choose featureless=False when defining a new GraphConvolution layer. However, the loss is still computed for each node, and I was wondering how I should change your code.
  • In the context of graph classification, how should I modify your train.py for batch-wise training?

Thanks for putting this together!