A PyTorch implementation of GraphSAGE

This package contains a PyTorch implementation of GraphSAGE. Currently, only supervised versions of GraphSAGE-mean, GraphSAGE-GCN, GraphSAGE-maxpool and GraphSAGE-meanpool are implemented.

Authors of this code package:

Bin Yu

Environment settings

  • python>=3.6.8
  • pytorch>=1.0.0

Basic Usage

Example Usage

To run the supervised model on Cuda:

python train.py


View Github