mirrors / dmlc--dgl

项目文件夹

文件

T

文件历史

Andrei Ivanov 1e3e98cd43 Improving the GATv2 example. (#6035 )

Co-authored-by: Hongzhi (Steve), Chen <chenhongzhi.nkcs@gmail.com>

2023-08-10 01:58:48 +00:00

..

gatv2.py

[Misc] Black auto fix. (#4642 )

2022-09-26 21:47:11 +08:00

README.md

[Example][Minor] Update README/requirements from sklearn -> scikit-learn (#5792 )

2023-06-20 07:47:55 +00:00

train.py

Improving the GATv2 example. (#6035 )

2023-08-10 01:58:48 +00:00

README.md

Graph Attention Networks v2 (GATv2)

Paper link: How Attentive are Graph Attention Networks?
Author's code repo: https://github.com/tech-srl/how_attentive_are_gats.
Annotated implemetnation: [https://nn.labml.ai/graphs/gatv2/index.html]

Dependencies

torch
requests
scikit-learn

How to run

Run with following:

python3 train.py --dataset=cora

python3 train.py --dataset=citeseer

python3 train.py --dataset=pubmed

Results

Dataset	Test Accuracy
Cora	82.10
Citeseer	70.00
Pubmed	77.2

All the accuracy numbers are obtained after 200 epochs.