Skip to content

Latest commit

 

History

History
60 lines (48 loc) · 1.36 KB

README.md

File metadata and controls

60 lines (48 loc) · 1.36 KB

Maximally Expressive GNNs for Outerplanar Graphs

Code repository for our paper Maximally Expressive GNNs for Outerplanar Graphs (GLF@NeurIPS 2023, oral).

Setup

Clone this repository and open the directory

Add this directory to the python path. Let $PATH be the path to where this repository is stored (i.e. the result of running pwd).

export PYTHONPATH=$PYTHONPATH:$PATH

Create a conda environment (this assume miniconda is installed)

conda create --name GNNs

Activate environment

conda activate GNNs

Install dependencies

conda install pytorch==1.12.0 torchvision==0.13.0 torchaudio==0.12.0 -c pytorch
conda install -c pyg pyg=2.2.0
python -m pip install -r requirements.txt

Replicating the experiments

Results can be found in the results directory.

Baselines:

bash Scripts/experiments_GIN_baselines.sh
bash Scripts/experiments_GCN_baselines.sh
bash Scripts/experiments_GAT_baselines.sh

CAT models:

bash Scripts/experiments_GIN_cat.sh
bash Scripts/experiments_GCN_cat.sh
bash Scripts/experiments_GAT_cat.sh

Benchmakr GIN vs CAT+GIN runtime:

bash Scripts/benchmark_training.sh

Benchmark CAT pre-processing time (results in terminal):

python Scripts/benchmark_cat.py

Compute directed effective resistance for CAT:

python Exp/resistance.py