Branch: master
Find file History
sufeidechabei and jermainewang [Model]PPI dataloader and inductive learning script. (#395)
* Create ppi.py

* Create train_ppi.py

* Update train_ppi.py

* Update train_ppi.py

* Create gat.py

* Update train.py

* Update train_ppi.py

* Update ppi.py

* Update train_ppi.py

* Update ppi.py

* Update train_ppi.py

* Update train_ppi.py

* Update ppi.py

* Update train_ppi.py

* update docs and readme
Latest commit 788d8dd Feb 18, 2019

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:

python train.py --dataset=cora --gpu=0
python train.py --dataset=citeseer --gpu=0
python train.py --dataset=pubmed --gpu=0 --num-out-heads=8 --weight-decay=0.001
python 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