Skip to content

Latest commit

 

History

History
 
 

gatv2

Folders and files

NameName
Last commit message
Last commit date

parent directory

..
 
 
 
 
 
 

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.