运行结果:
官方代码:https://github.com/THUDM/GATNE 效果:
Overall ROC-AUC: 0.912732657002574
Overall PR-AUC 0.8709920664369216
Overall F1: 0.8447942506931271
本代码效果:
Overall ROC-AUC: 0.9336145414474002
Overall PR-AUC 0.9088397518884685
Overall F1: 0.8619035458923099