dmlc--dgl
Israt Nisa
765d725e15
* GAT example * Complete GAT and Add README * Update Co-authored-by: Israt Nisa <nisisrat@amazon.com> Co-authored-by: czkkkkkk <zekucai@gmail.com>
Graph Attention Networks (GAT)
- Paper link: https://arxiv.org/abs/1710.10903
- Author's code repo (tensorflow implementation): https://github.com/PetarV-/GAT.
- Popular pytorch implementation: https://github.com/Diego999/pyGAT.
How to run
Sparse tensor GATConv module
Run with the following (available dataset: "cora", "citeseer", "pubmed")
python3 train.py --dataset cora
Summary
- cora: ~0.810
- citeseer: ~0.697
- pubmed: ~0.774