Skip to content

Latest commit

 

History

History
 
 

gaan

Folders and files

NameName
Last commit message
Last commit date

parent directory

..
 
 
 
 

Gated Attention Networks (GaAN)

Dataset Statics

Dataset # Nodes # Edges # Classes
Cora 2,708 10,556 7
Citeseer 3,327 9,228 6
Pubmed 19,717 88,651 3

Refer to Planetoid.

Results

TL_BACKEND="tensorflow" python GaAN_trainer.py --dataset cora --n_epoch 200 --lr 0.005 --l2_coef 0.01 --drop_rate 0.1 --heads 8 --v 64 --m 64
TL_BACKEND="tensorflow" python GaAN_trainer.py --dataset citeseer --n_epoch 200 --lr 0.003 --l2_coef 0.005 --drop_rate 0.4 --heads 8 --v 32 --m 32
TL_BACKEND="tensorflow" python GaAN_trainer.py --dataset pubmed --n_epoch 300 --lr 0.005 --l2_coef 0.0005 --drop_rate 0.4 --heads 8 --v 64 --m 64
Dataset GAT(Paper) Our(tf)
cora 83.0(±0.7) 79.11(±1.78)
citeseer 72.5(±0.7) 68.15(±3.75)
pubmed 79.0(±0.3) 77.29(±1.71)