- Paper link: https://arxiv.org/abs/2201.05540
- Author's code repo: https://github.com/liun-online/CoGSL.
| Dataset | # Nodes | # Edges | # Classes |
|---|---|---|---|
| Polblogs | 1222 | 33428 | 2 |
| Citeseer | 3,327 | 9,228 | 6 |
| Wiki-CS | 11701 | 291039 | 10 |
| MS Academic | 18333 | 163788 | 15 |
# available dataset: "polblogs", "citeseer", "wikics", "ms"
TL_BACKEND="paddle" python cogsl_trainer.py --dataset polblogs --v1_p 1 --v2_p 300 --cls_hid 16 --gen_hid 64 --mi_hid 128
TL_BACKEND="paddle" python cogsl_trainer.py --dataset citeseer --v1_p 2 --v2_p 40 --cls_hid 16 --gen_hid 32 --mi_hid 128
TL_BACKEND="paddle" python cogsl_trainer.py --dataset wikics --v1_p 1 --v2_p 1 --cls_hid 16 --gen_hid 16 --mi_hid 32 -ggl True --big True
TL_BACKEND="paddle" python cogsl_trainer.py --dataset ms --v1_p 1 --v2_p 1 --cls_hid 16 --gen_hid 32 --mi_hid 256 -ggl True --big True
TL_BACKEND="torch" python cogsl_trainer.py --dataset polblogs --v1_p 1 --v2_p 300 --cls_hid 16 --gen_hid 64 --mi_hid 128
TL_BACKEND="torch" python cogsl_trainer.py --dataset citeseer --v1_p 2 --v2_p 40 --cls_hid 16 --gen_hid 32 --mi_hid 128
TL_BACKEND="torch" python cogsl_trainer.py --dataset wikics --v1_p 1 --v2_p 1 --cls_hid 16 --gen_hid 16 --mi_hid 32 -ggl True --big True
TL_BACKEND="torch" python cogsl_trainer.py --dataset ms --v1_p 1 --v2_p 1 --cls_hid 16 --gen_hid 32 --mi_hid 256 --ggl True --big True
| 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 |