Skip to content

Latest commit

 

History

History
26 lines (17 loc) · 619 Bytes

README.MD

File metadata and controls

26 lines (17 loc) · 619 Bytes

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