Skip to content
This repository has been archived by the owner on Nov 10, 2023. It is now read-only.
/ SAGPool Public archive

Official PyTorch Implementation of SAGPool - ICML 2019

Notifications You must be signed in to change notification settings

inyeoplee77/SAGPool

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

7 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

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}
}

About

Official PyTorch Implementation of SAGPool - ICML 2019

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages