The final code for OGB
Installation Requirements:
- ogb=1.3.1
- torch=1.7.0
- torch-geometric=1.7.0
- torch-scatter=2.0.6
- torch-sparse=0.6.9
Baseline models
- The path of ogbn-ppa dataset should be ogb/ppa/dataset/ogbn_ppa, then you can get the performance reported by running the default code 10 times. In fact, we can get 82 by other parameters, but the model will be too large and training is much more slower. So in this report, we only set the parameters for the performance 81.4.
- The path of ogbn-code2 dataset should be ogb/code/dataset/ogbn_code2. Then you can get the performance reported by running the default code 10 times
ogbg-ppa
- cd ogb/ppa;
- sh run_script.sh 0
ogbg-code2
- cd ogb/code;
- sh run_script.sh 0
Performance
bot : bag of tricks
ogbg-ppa
methods | test accuracy | val accuracy | Parameters | Hardware |
---|---|---|---|---|
ExpC+bot | 0.8140±0.0028 | 0.7811±0.0012 | 3.7M | Tesla V100 32GB |
ogbg-code2
methods | test accuracy | val accuracy | Parameters | Hardware |
---|---|---|---|---|
GMAN+bot | 0.1770±0.0012 | 0.1631±0.0090 | 63.7M | Tesla V100 32GB |