CIMAGE: Exploiting Conditional Independence in Masked Graph Auto-encoders

This repository is an official PyTorch implementation of CIMAGE.

📌 Dependencies

Please refer to the environment.yaml file for the required library dependencies.

🚀 Getting Started

📊 Node Classification

To train the model for node classification, run the following command:

python train_nodeclas.py --dataset Photo

🔗 Link Prediction

To train the model for link prediction, use:

python train_linkpred.py --dataset Cora

🔍 Reproducibility

Logs containing accuracy and loss metrics for each epoch are included to facilitate reproduction. If you encounter any challenges in reproducing the results, kindly refer to these logs for guidance