GPA

February 11, 2022 · View on GitHub

This is a pytorch and pytorch-geometric based implementation of Graph Contrastive Learning with Personalized Augmentation.

Installation

The required packages can be installed by running pip install -r requirements.txt.

The datasets used in our paper can be automatically downlowad.

Train on the TUDatasets (PROTEINS, NCI1, DD, COLLAB, MUTAG, IMDB-BINARY, REDDIT-BINARY, REDDIT-MULTI-5K, github_stargazers):

python main_gpa.py --dataset "MUTAG" --mode unsuper
python main_gpa.py --dataset "MUTAG" --mode semi