zhengyu19921215 / graphsage-diffpool-classifier

Implemented a GNN for classification tasks in PyTorch, combining Graph Convolutions and Graph Pooling layers

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