项目文件夹

文件

T

文件历史

Quan (Andy) Gan 6c23fba8fa [Doc][Org] move edge_softmax to functional (#2442 )

* move edge_softmax to functional

* change examples

* fixes

* revert

* fix

* fix

* remove

* fix

* test

* add init

Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com>

2021-01-26 12:46:37 +08:00

gat.py

[Doc][Org] move edge_softmax to functional (#2442 )

2021-01-26 12:46:37 +08:00

README.md

[Dataset] Citation graph (#1902 )

2020-08-01 19:25:50 +08:00

train.py

[Dataset] Citation graph (#1902 )

2020-08-01 19:25:50 +08:00

utils.py

[Tensorflow] Several nn & example (#1191 )

2020-01-19 22:48:40 +08:00

README.md

Graph Attention Networks (GAT)

Paper link: https://arxiv.org/abs/1710.10903
Author's code repo (in Tensorflow): https://github.com/PetarV-/GAT.
Popular pytorch implementation: https://github.com/Diego999/pyGAT.

Dependencies

tensorflow 2.1.0+
requests

pip install tensorflow requests
DGLBACKEND=tensorflow

How to run

Run with following:

python3 train.py --dataset=cora --gpu=0

python3 train.py --dataset=citeseer --gpu=0 --early-stop

python3 train.py --dataset=pubmed --gpu=0 --num-out-heads=8 --weight-decay=0.001 --early-stop

Results

Dataset	Test Accuracy	Baseline (paper)
Cora	84.2	83.0(+-0.7)
Citeseer	70.9	72.5(+-0.7)
Pubmed	78.5	79.0(+-0.3)

All the accuracy numbers are obtained after 200 epochs.
All time is measured on EC2 p3.2xlarge instance w/ V100 GPU.