mirrors / dmlc--dgl

项目文件夹

文件

T

Shaked Brody e2f33fd5cc [NN][Model] GATv2 (#3473 )

* [Model][Core] GATv2

* lint

* gatv2conv.py

* lint

* lint

* style and docs

* lint

* gatv2conv fix

Co-authored-by: Shaked Brody shakedbr@campus.technion.ac.il <shakedbr@tangerine.cslcs.technion.ac.il>
Co-authored-by: Mufei Li <mufeili1996@gmail.com>

2021-11-03 21:55:56 +08:00

812 B

原始文件 永久链接 Blame 文件历史

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
sklearn

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.