- All model checkpoints can be found in
./ckpts
- Here's the link for downloading the processed datasets.
- After downloading the datasets zip, please
unzipit in the root folder. - To run the additional experiments on the REDDIT-BINARY dataset, download this dataset from here and unzip in the root folder like the other datasets.
Codes in this repo have been tested on python3.10 + pytorch2.1 + pyg2.5.
To reproduce the exact python environment, please run:
conda create -n gnnboundary poetry jupyter
conda activate gnnboundary
poetry installOnce the environment is installed and the datasets are set up, our experiments can be easily reproduced by running the run_experiments.py script.
To reproduce the results for all datasets and models, run:
python run_experiments.py motif
python run_experiments.py collab
python run_experiments.py enzymes
python run_experiments.py binary_reddit
python run_experiments.py motif_gat