项目文件夹

文件

T

文件历史

Hongzhi (Steve), Chen a9f2acf323 [Misc] Black auto fix. (#4641 )

* [Misc] Black auto fix.

* sort

Co-authored-by: Steve <ubuntu@ip-172-31-34-29.ap-northeast-1.compute.internal>

2022-09-26 21:46:32 +08:00

gat.py

[Misc] Black auto fix. (#4641 )

2022-09-26 21:46:32 +08:00

README.md

[Hotfix] Fix GAT example and clarify the usage of early stop (#1065 )

2019-11-30 23:43:15 +08:00

train.py

[Misc] Black auto fix. (#4641 )

2022-09-26 21:46:32 +08:00

utils.py

[Misc] Black auto fix. (#4641 )

2022-09-26 21:46:32 +08:00

README.md

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