yusonghust / graphsage_tf

Tensorflow implementation of 'Inductive Representation Learning on Large Graphs'

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Graphsage

Tensorflow implementation of 'Inductive Representation Learning on Large Graphs'

Introduction

A tensorflow re-implementation of graphsage, which is easier than the original implementation GraphSAGE original implementation.
This code includes supervised and uinsupervised version, and three types of aggregators('mean','pooling' and 'lstm').

Requirement

python 3.6, tensorflow 1.12.0

Usage

To see and modify the parameters of graphsage, see config.py.
To run the codes, use:

python main.py

Results

Here shows accuracy of the supervised and unsupervised graphsage with 'mean' aggregator.

The supervised graphsage accuracy is 0.871

supervised accuracy=0.871

The unsupervised graphsage accuracy is 0.790

unsupervised accuracy=0.0.79

About

Tensorflow implementation of 'Inductive Representation Learning on Large Graphs'


Languages

Language:Python 100.0%