sunfanyunn / InfoGraph

Official code for "InfoGraph: Unsupervised and Semi-supervised Graph-Level Representation Learning via Mutual Information Maximization" (ICLR 2020, spotlight)
https://openreview.net/forum?id=r1lfF2NYvH
312 stars 45 forks source link

Gap between the performance reported and that of this repo on QM9. #8

Closed hengruizhang98 closed 3 years ago

hengruizhang98 commented 3 years ago

Hi, we notice that there is a gap between the performance on the QM9 dataset reported and that we reproduced using your code. e.g. for HOMO(2), the reported RMSE is 0.0060, while with your code it's 0.1560.

sunfanyunn commented 3 years ago

I conducted those experiments for the paper in 2019 and the version for pytorch-geometric==1.3.0 at the time (even though I later tested the code in this repo with pytorch-geometric==1.6.0) Apparently the code and source of QM9 dataset have been changed. Refer to

sunfanyunn commented 3 years ago

consider the performance reported by this paper in 2019: https://rlgm.github.io/papers/31.pdf

hengruizhang98 commented 3 years ago

Got it, thank you very much!