Why could I get node representation only by GIN?

zaixizhang / MGSSL

Official implementation of NeurIPS'21 paper"Motif-based Graph Self-Supervised Learning for Molecular Property Prediction"

Apache License 2.0

111 stars 16 forks source link

Why could I get node representation only by GIN? #15

Open simmzx opened 9 months ago

simmzx commented 9 months ago

Hello, Zhang: When I run the ''model.py'' , only "gin_type" = "gin" work successfully, and other models, including gat, gcn, graphsage and attentivefp are fail to work. All of them were base on your code. I am appreciate you will resoponse to me.

The traceback file is attached. [Uploading traceback.do traceback.docx cx…]()

Best regard, Xiang Zhang

zaixizhang commented 8 months ago

Hi Xiang,

Thanks for your interest in our work! In the github repo, we only provide the checkpoint for the gin model. This may be reason why it fails with other models such as gat and gcn.

Bests,