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=Falsewhen defining a newGraphConvolutionlayer. 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.pyfor batch-wise training?
Thanks for putting this together!
Source: tkipf/gcn