This is the course assignment of the course "Deep learning algorithm design". This repository is the implement of the code on graph classification with deep learning, and we use the graph pool method. The paper refers to "Self-Attention Graph Pooling"

The model reference paper on 2018 ICLR : Self-Attention Graph Pooling

requirement on windows 10

pytorch 1.6.0
pytorch-geometric 1.6.1