Skip to content

Latest commit

 

History

History
22 lines (13 loc) · 472 Bytes

README.md

File metadata and controls

22 lines (13 loc) · 472 Bytes

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