Question about data augmentation

[Struggle-Forever]

Question about data augmentation: When subgraph augmentation is performed, essentially only the enge_index is modified. This means that only the subgraph is aggregated for information propagation. What puzzles me is that when computing the graph level features, the pooling function involves the features of all nodes of the graph (selected subgraph nodes and unselected nodes), does this make sense? Isn't the pooling at this point only considering subgraphs?

[yyou1996]

Hi @Struggle-Forever,

Please refer to an instantiation of the subgraph augmentation #24 (comment). We would drop the non-selected nodes via, e.g. data.x = data.x[idx_nondrop] where the pooling function is operated on.