Skip to content

Latest commit

 

History

History
 
 

cogsl

Folders and files

NameName
Last commit message
Last commit date

parent directory

..
 
 
 
 

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