Graph Convolution Network for PyTorch

Pytorch 1.0 with python 3.7.

GCN implementation for paper: Semi-Supervised Classification with Graph Convolutional Networks

Benchmark

dataset	Citeseea	Cora	Pubmed	NELL
GCN(official)	70.3	81.5	79.0	66.0
This repo.		81.7	78.7

Hints: To achieve 81.7% on Cora dataset, pytorch need 400 epochs. and it train relatively slower on Pubmed dataset.

HOWTO

python train.py

Screenshot