Skip to content
Official PyTorch Implementation of SAGPool - ICML 2019
Branch: master
Clone or download
Latest commit 7199d7f Jun 22, 2019
Permalink
Type Name Latest commit message Commit time
Failed to load latest commit information.
.gitignore pytorch_geometric Jun 2, 2019
README.md readme Jun 22, 2019
figure.png
layers.py pytorch_geometric Jun 2, 2019
main.py pytorch_geometric Jun 2, 2019
networks.py pytorch_geometric Jun 2, 2019

README.md

Pytorch implementation of Self-Attention Graph Pooling

PyTorch implementation of Self-Attention Graph Pooling

SAGPool

Requirements

  • torch_geometric
  • torch

Usage

python main.py

Cite

@InProceedings{pmlr-v97-lee19c,
  title = 	 {Self-Attention Graph Pooling},
  author = 	 {Lee, Junhyun and Lee, Inyeop and Kang, Jaewoo},
  booktitle = 	 {Proceedings of the 36th International Conference on Machine Learning},
  year = 	 {2019},
  month = 	 {09--15 Jun}
}
You can’t perform that action at this time.