rampasek / GraphGPS

Recipe for a General, Powerful, Scalable Graph Transformer
MIT License
643 stars 114 forks source link

SBM (PATTERN and CLUSTER) accuracy fix #12

Closed rampasek closed 1 year ago

rampasek commented 1 year ago

Accuracy for SBM-derived datasets PATTERN and CLUSTER is supposed to be reweighed by class. This was not the case in GraphGPS code before and this PR fixes it.

New GPS performance:

PATTERN: 86.685 ± 0.059 (decreased from original 90.324 ± 0.132) CLUSTER: 78.016 ± 0.180 (increased from original 77.95 ± 0.305)