Skip to content

A reimplementation of semi-supervised-classification with GCN

Notifications You must be signed in to change notification settings

FabianPieroth/PyTorch-GCN-cora

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

12 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

PyTorch-GCN-cora

A reimplementation of the paper Semi-Supervised Classification with Graph Convolutional Networks by Kipf and Welling 2016 in PyTorch. See also their official implementation here. I will solely focus on reproducing the results for the cora dataset, which I pushed into the git repository as it is not very large.

The early stopping class is taken from the repository of Bjarten. I only changed the process of saving the model slightly.

Setup

Run:

make environment 
source activate pytorch-gcn
make requirements

About

A reimplementation of semi-supervised-classification with GCN

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published