Skip to content
Graph Filter Neural Network
Python
Branch: master
Clone or download
Latest commit 2daf25b May 30, 2019
Permalink
Type Name Latest commit message Commit time
Failed to load latest commit information.
LICENSE make gfnn public May 26, 2019
README.md Update README.md May 31, 2019
args.py Update args.py May 31, 2019
citation.py fix gfnn name May 30, 2019
metrics.py make gfnn public May 26, 2019
models.py fix gfnn name May 30, 2019
noise.py make gfnn public May 26, 2019
normalization.py make gfnn public May 26, 2019
raw_feats_citation.py make gfnn public May 26, 2019
reddit.py make gfnn public May 26, 2019
requirements.txt make gfnn public May 26, 2019
synthetic.py fix gfnn name May 30, 2019
synthetic_data.py make gfnn public May 26, 2019
train.py make gfnn public May 26, 2019
transformation.py
tuning.py make gfnn public May 26, 2019
utils.py make gfnn public May 26, 2019

README.md

Revisiting Graph Neural Networks: All We Have is Low-Pass Filters

In this work, we study graph neural networks for vertex classification. This work is conducted at RIKEN AIP. Our preprint is uploaded on arXiv.

Requirements

Create an Python 3.6 environment and install these packages:

numpy
scipy
networkx==1.11
scikit-learn
pytorch
torchvision

Or install from requirements.txt.

Move files in data/* to your ~/data/gcn/ folder, or change the paths in utils.py for Cora, Citeseer, Pubmed. The data files can be found here (Gdrive).

For Reddit, download reddit.adj (Gdrive) and reddit.npz (Gdrive) and put them under ~/data/reddit/, or change paths in utils.py.

We will pack data for PPI in the later version.

Experiments

Check args.py file to change options for noise and other filters.

To run with Cora, Citeseer, Pubmed:

python citation.py --no-cuda --model gfnn --dataset cora
python citation.py --no-cuda --model SGC --dataset cora
python citation.py --no-cuda --model GCN --dataset cora

To run with synthetic dataset (4000 data points):

python synthetic.py --no-cuda --model gfnn
python synthetic.py --no-cuda --model SGC
python synthetic.py --no-cuda --model GCN

The frequency experiment (Figure 3) can be found in transformation.py and raw_feats_citation.py.

Acknowledgement

This reposipory is built upon SGC, FastGCN, and GCN. I would like to say thanks to the authors of these repositories for making their code available.

You can’t perform that action at this time.