Official Code Repository for the paper "Teacher-Guided Graph Contrastive Learning" (TMLR 2024)

This repository implements Teacher-Guided Graph Contrastive Learning (TGCL).

To download the molecular graph dataset, run commands below

wget http://snap.stanford.edu/gnn-pretrain/data/chem_dataset.zip
unzip chem_dataset.zip -d transferLearning_MoleculeNet_PPI/chem/

To download the biological graph dataset, run commands below

wget http://snap.stanford.edu/gnn-pretrain/data/bio_dataset.zip
unzip bio_dataset.zip -d transferLearning_MoleculeNet_PPI/bio/

Clone D-SLA and follow the instructions to train the D-SLA teacher

Clone GraphLog and follow the instructions to train the GraphLog teacher

Run pretrain_job.sh

Run ft_job.sh