Awesome
GiGaMAE
This repository hosts the code for our CIKM'23 long paper 'GiGaMAE: Generalizable Graph Masked Autoencoder via Collaborative Latent Space Reconstruction.' by Yucheng Shi, Yushun Dong, Qiaoyu Tan, Jundong Li, and Ninghao Liu.
Dependencies
- torch 1.12.0+cu116
- torch-cluster 1.6.0
- torch-geometric 2.1.0
- torch-scatter 2.0.9
- torch-sparse 0.6.15
If you have trouble in installing torch-geometric
, you may find help in its official website.
Training & Evaluation
Node Classification and Clustering
sh run.sh
Link Prediction
sh run_lp.sh
Acknowledgements
Parts of implementation are reference to GRACE