Skip to content

Representation learning on large graphs using stochastic graph convolutions.

Notifications You must be signed in to change notification settings

bkj/pytorch-graphsage

Folders and files

NameName
Last commit message
Last commit date
May 8, 2018
May 8, 2018
Nov 2, 2017
Oct 17, 2017
Nov 17, 2017
Oct 17, 2017
Nov 2, 2017
Nov 2, 2017
Nov 2, 2017
Nov 2, 2017
Nov 2, 2017

Repository files navigation

graphsage

pytorch version of https://github.com/williamleif/GraphSAGE

There are also a handful of new features, including:

  • scripts for preprocessing data
  • attention-based aggregator
  • sparse edge sampler (eg, don't use the dense 2D edgelist)
  • richer, pluggable preprocessing classes

This version is missing some features, as well. Namely:

  • we don't support unsupervised models (yet)

A simplified version of the original Tensorflow version can be found on branch tf-simple.

LICENSE

MIT