/FatraGNN

Primary LanguagePythonMIT LicenseMIT

FatraGNN

This repository (without datasets due to space limit) is the official implementation of Fairness Learning on Graphs under Distribution Shifts.

Requirements

To install requirements:

pip install -r requirements.txt

Training

To train FatraGNN on the five datasets, run these commands:

python fatragnn.py --gpu 3 --ood 2 --dataset=bail --encoder=GCN --inid=_B0 --outid=all --times=config1
python fatragnn.py --gpu 3 --ood 2 --dataset=credit --encoder=GCN --inid=_C0 --outid=all --times=config1
python fatragnn.py --gpu 3 --ood 2 --dataset=pokec --encoder=GCN --inid=_z --outid=all --times=config1

Train on sync-B1s and sync-B2s

python fatragnn.py --gpu 3 --ood 1 --dataset=bail --encoder=GCN --inid=_B0 --outid=_md0 --times=config2
python fatragnn.py --gpu 3 --ood 1 --dataset=bail --encoder=GCN --inid=_B0 --outid=_md3 --times=config3