zfx233 / CMGNN

The codebase of "Revisiting the Message Passing in Heterophilous Graph Neural Networks", including 13 baseline methods and proposed CMGNN.
2 stars 0 forks source link

CMGNN

This is a unified codebase for heterophilous graph neural networks, including 13 baseline methods, one novel method CMGNN, and 10 datasets.

Baseline Methods

Datasets

The datasets used in the codebase include Roman-Empire, Amazon-Ratings, Chameleon-F, Squirrel-F, Actor, Flickr, BlogCatalog, Wikics, Pubmed, and Photo.

How to Run

Baseline Methods

The presearched parameters are listed in '/config/baseline/{model_type}.yaml'.

python main.py --model_type={model_type} --dataset={dataset}

Note the dataset name is the script should use plain lowercase letters, e.g. for Chameleon-F dataset with "--dataset=chameleonf".

Compatibility Matrix-aware GNN (CMGNN)

python main.py --model_type=CMGNN --dataset={dataset}

Main Requirements