Skip to content

Latest commit

 

History

History

gen

Folders and files

NameName
Last commit message
Last commit date

parent directory

..
 
 
 
 

Graph Structure Estimation Neural Networks (GEN)

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

# available dataset: "cora", "citeseer", "pubmed"
TL_BACKEND="paddle" python gen_trainer.py --dataset cora
TL_BACKEND="paddle" python gen_trainer.py --dataset citeseer
TL_BACKEND="tensorflow" python gen_trainer.py --dataset cora 
TL_BACKEND="tensorflow" python gen_trainer.py --dataset citeseer
TL_BACKEND="torch" python gen_trainer.py --dataset cora
TL_BACKEND="torch" python gen_trainer.py --dataset citeseer
Dataset Paper Our(pd) Our(tf) Our(th)
cora 83.6±0.4 82.60 82.50
citeseer 73.8±0.6 71.50 69.50