jerry2398 / graphprompt

GraphPrompt: Unifying Pre-Training and Downstream Tasks for Graph Neural Networks

We provide the code (in pytorch) and datasets for our paper "GraphPrompt: Unifying Pre-Training and Downstream Tasks for Graph Neural Networks", which is accepted by WWW2023.

Description

The repository is organised as follows:

• data/: contains data we use.
• graphdownstream/: implements pre-training and downstream tasks at the graph level.
• nodedownstream/: implements downstream tasks at the node level.
• convertor/: generate raw data.

Package Dependencies

• cuda 11.3
• dgl0.9.0-cu113
• dgllife

Running experiments

Graph Classification

Default dataset is ENZYMES. You need to change the corresponding parameters in pre_train.py and prompt_fewshot.py to train and evaluate on other datasets.

Pretrain:

• python pre_train.py

Prompt tune and test:

• python prompt_fewshot.py

Node Classification

Default dataset is ENZYMES. You need to change the corresponding parameters in prompt_fewshot.py to train and evaluate on other datasets.

Prompt tune and test:

• python run.py

Citation

@inproceedings{liu2023graphprompt,
title={GraphPrompt: Unifying Pre-Training and Downstream Tasks for Graph Neural Networks},
author={Liu, Zemin and Yu, Xingtong and Fang, Yuan and Zhang, Xinming},
booktitle={Proceedings of the ACM Web Conference 2023},
year={2023}
}