https://github.com/inyeoplee77/SAGPool

Official PyTorch Implementation of SAGPool - ICML 2019
https://github.com/inyeoplee77/SAGPool

Last synced: about 1 month ago
JSON representation

Official PyTorch Implementation of SAGPool - ICML 2019

• Host: GitHub
• URL: https://github.com/inyeoplee77/SAGPool
• Owner: inyeoplee77
• Archived: true
• Created: 2019-05-13T16:33:57.000Z (over 5 years ago)
• Default Branch: master
• Last Pushed: 2023-11-10T12:46:48.000Z (about 1 year ago)
• Last Synced: 2024-08-01T17:24:56.084Z (4 months ago)
• Language: Python
• Homepage:
• Size: 90.8 KB
• Stars: 358
• Watchers: 8
• Forks: 80
• Open Issues: 10
• Metadata Files:
  • Readme: README.md

Awesome Lists containing this project

• awesome-graph-classification - [Python Reference
• StarryDivineSky - inyeoplee77/SAGPool - Attention Graph Pooling torch 自我注意力图池化 (图注意力机制 / 网络服务_其他)

README

        
Pytorch implementation of Self-Attention Graph Pooling

====

PyTorch implementation of [Self-Attention Graph Pooling](https://arxiv.org/abs/1904.08082)

![SAGPool](figure.png)

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

}

```