koshinryuu / keras-gcn

Keras implementation of Graph Convolutional Networks

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

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

About

Keras implementation of Graph Convolutional Networks

License:MIT License


Languages

Language:Python 100.0%