Skip to content
Branch: master
Find file History
Fetching latest commit…
Cannot retrieve the latest commit at this time.
Permalink
Type Name Latest commit message Commit time
..
Failed to load latest commit information.
README.md
gat.py
train.py
train_ppi.py

README.md

Graph Attention Networks (GAT)

Dependencies

  • torch v1.0: the autograd support for sparse mm is only available in v1.0.
  • requests
  • sklearn
pip install torch==1.0.0 requests

How to run

Run with following:

python3 train.py --dataset=cora --gpu=0
python3 train.py --dataset=citeseer --gpu=0
python3 train.py --dataset=pubmed --gpu=0 --num-out-heads=8 --weight-decay=0.001
python3 train_ppi.py --gpu=0

Results

Dataset Test Accuracy Time(s) Baseline#1 times(s) Baseline#2 times(s)
Cora 84.0% 0.0127 0.0982 (7.7x) 0.0424 (3.3x)
Citeseer 70.7% 0.0123 n/a n/a
Pubmed 78.1% 0.0302 n/a n/a
You can’t perform that action at this time.