Skip to content

Latest commit

 

History

History
 
 

dropedge

Folders and files

NameName
Last commit message
Last commit date

parent directory

..
 
 
 
 

DropEdge

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(cora & 4 layer GCN)

# available dataset: "cora", "citeseer", "pubmed"
TL_BACKEND="paddle" python dropedge_trainer.py --dataset cora --lr 0.01 --l2_coef 0.0005 --drop_rate 0.8 --n_layers 4 --sampling_percent 0.7
TL_BACKEND="paddle" python dropedge_trainer.py --dataset cora --lr 0.01 --l2_coef 0.0005 --drop_rate 0.8 --n_layers 4 --sampling_percent 1
TL_BACKEND="tensorflow" python dropedge_trainer.py --dataset cora --lr 0.01 --l2_coef 0.0005 --drop_rate 0.8 --n_layers 4 --sampling_percent 0.7
TL_BACKEND="tensorflow" python dropedge_trainer.py --dataset cora --lr 0.01 --l2_coef 0.0005 --drop_rate 0.8 --n_layers 4 --sampling_percent 1
TL_BACKEND="torch" python dropedge_trainer.py --dataset cora --lr 0.01 --l2_coef 0.0005 --drop_rate 0.8 --n_layers 4 --sampling_percent 0.7
TL_BACKEND="torch" python dropedge_trainer.py --dataset cora --lr 0.01 --l2_coef 0.0005 --drop_rate 0.8 --n_layers 4 --sampling_percent 1
Method Paper Our(tf) Our(th) Our(pd)
Original 80.40 80.42±1.2 81.16±1.14 80.44±0.66
DropEdge 82.00 81.18±2.22 81.94±1.56 81.20±1.4
  • The model performance is the average of 5 tests