Skip to content

Latest commit

 

History

History
24 lines (20 loc) · 809 Bytes

README.md

File metadata and controls

24 lines (20 loc) · 809 Bytes

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