gat

Graph Attention Networks (GAT)

• Paper link: https://arxiv.org/abs/1710.10903
• Author's code repo: https://github.com/PetarV-/GAT.

Note that the original code is implemented with Tensorflow for the paper.

Dependencies

• MXNet nightly build
• requests

pip install mxnet --pre
pip install requests

Usage (make sure that DGLBACKEND is changed into mxnet)

DGLBACKEND=mxnet python3 train.py --dataset cora --gpu 0
DGLBACKEND=mxnet python3 train.py --dataset citeseer --gpu 0 --early-stop
DGLBACKEND=mxnet python3 train.py --dataset pubmed --gpu 0 --early-stop