SAGPool

使用基于自注意力池化机制结合GCN模型实现图分类

所需环境：

1.torch:1.4.0
2.torch_scatter:2.0.3
3.scipy:1.4.1

数据集

1.DD:https://ls11-www.cs.tu-dortmund.de/people/morris/graphkerneldatasets/DD.zip
2.COX2:https://ls11-www.cs.tu-dortmund.de/people/morris/graphkerneldatasets/COX2.zip
3.BZR:https://ls11-www.cs.tu-dortmund.de/people/morris/graphkerneldatasets/BZR.zip

运行方式

1.下载整个项目
2.定位到self_attn_pool.py位置，运行pyhton self_attn_pool.py