benjamesbabala/keras-gcn

Keras implementation of Graph Convolutional Networks

Deep Learning on Graphs with Keras

Keras-based implementation of graph convolutional networks for semi-supervised classification [1]. This also implements filters from [2] and makes use of the Cora dataset from [3].

Installation

python setup.py install

Dependencies

• keras (1.0.9 or higher)
• TensorFlow or Theano

Usage

python train.py

References

[1] Kipf & Welling, Semi-Supervised Classification with Graph Convolutional Networks, 2016

[2] Defferrard et al., Convolutional Neural Networks on Graphs with Fast Localized Spectral Filtering, NIPS 2016

[3] Sen et al., Collective Classification in Network Data, AI Magazine 2008