Graph Convolution + Pooling NN PyTorch Implementation

Generating a graph classification NN in PyTorch, combining Graph Convolutions and Graph Pooling modules

Work based on:

• GCNs
• GraphSAGE
• DiffPool
• Graph U-net