Skip to content

Latest commit

 

History

History
38 lines (29 loc) · 2.03 KB

readme.md

File metadata and controls

38 lines (29 loc) · 2.03 KB

Compact Graph Structure Learning via Mutual Information Compression (CoGSL)

Dataset Statics

Dataset # Nodes # Edges # Classes
Polblogs 1222 33428 2
Citeseer 3,327 9,228 6
Wiki-CS 11701 291039 10
MS Academic 18333 163788 15

Results

# available dataset: "polblogs", "citeseer", "wikics", "ms"
TL_BACKEND="paddle" python cogsl_trainer.py polblogs
TL_BACKEND="paddle" python cogsl_trainer.py citeseer
TL_BACKEND="paddle" python cogsl_trainer.py wikics
TL_BACKEND="paddle" python cogsl_trainer.py ms
TL_BACKEND="torch" python cogsl_trainer.py polblogs
TL_BACKEND="torch" python cogsl_trainer.py citeseer
TL_BACKEND="torch" python cogsl_trainer.py wikics
TL_BACKEND="torch" python cogsl_trainer.py ms

Dataset Metric Paper Our(pd) Our(th)
polblogs F1-macro
F1-micro
AUC
95.5±0.1
95.5±0.1
98.3±0.1
96.0±0.2
96.0±0.2
98.7±0.0
95.2±0.4
95.2±0.4
98.2±0.1
citeseer F1-macro
F1-micro
AUC
70.2±0.6
73.4±0.8
91.4±0.5
69.0±0.3
72.4±0.5
91.1±0.2
64.3±0.5
67.3±0.5
88.0±0.3
wikics F1-macro
F1-micro
AUC
72.3±0.6
75.0±0.3
96.4±0.2
71.3±0.4
74.2±0.3
95.3±0.3
65.1±0.5
69.0±0.8
93.5±0.4
ms F1-macro
F1-micro
AUC
90.5±0.4
92.4±0.5
99.4±0.1
88.1±0.0
90.1±0.0
98.9±0.0
77.8±0.0
82.9±0.0
98.0±0.0