An open API service indexing awesome lists of open source software.

https://github.com/aveek-saha/graph-conv-net

A TensorFlow 2 implementation of Graph Convolutional Networks (GCN)
https://github.com/aveek-saha/graph-conv-net

gcn gcnn graph-convolutional-networks graph-neural-networks graphs tensoflow tensorflow2

Last synced: 8 months ago
JSON representation

A TensorFlow 2 implementation of Graph Convolutional Networks (GCN)

Awesome Lists containing this project

README

          

# Graph Convolutional Networks
A TensorFlow 2 implementation of Graph Convolutional Networks for classification of nodes from the paper, Thomas N. Kipf, Max Welling, [Semi-Supervised Classification with Graph Convolutional Networks](http://arxiv.org/abs/1609.02907) (ICLR 2017)

This is my attempt at trying to understand and recreate the neural network from from the paper. You can find the official implementation here: https://github.com/tkipf/gcn

## Requirements
- tensorflow 2
- networkx
- numpy
- scikit-learn

## Run

To train and test the network with the CORA dataset.

```bash
python train.py
```

## Cite

Please cite the original paper if you use this code in your own work:

```
@inproceedings{kipf2017semi,
title={Semi-Supervised Classification with Graph Convolutional Networks},
author={Kipf, Thomas N. and Welling, Max},
booktitle={International Conference on Learning Representations (ICLR)},
year={2017}
}
```