What GNN method is used in this implementation?

[Yoyo0320]

Hi,

According to the paper, it said you used the GraphSAGE as the GNN module and stack two layers before diffpool layer. However, in this code implementation, are you using GCN instead and stack 3 layers before diffpool?

[RexYing]

The difference from GCN is the DeepSet aggregation, and concatenation in GraphSAGE. But you are right, that this is doing whole graph convolution, rather than the minibatch updates in GraphSAGE.