Skip to content

Latest commit

 

History

History
40 lines (30 loc) · 812 Bytes

README.md

File metadata and controls

40 lines (30 loc) · 812 Bytes

Graph Attention Networks v2 (GATv2)

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.