Skip to main content

Graph engine - distributed graph engine to host graphs.

Project description

DeepGNN Overview

DeepGNN is a framework for training machine learning models on large scale graph data. DeepGNN contains all the necessary features including:

  • Distributed GNN training and inferencing on both CPU and GPU.
  • Custom graph neural network design.
  • Online Sampling: Graph Engine (GE) will load all graph data, each training worker will call GE to get node/edge/neighbor features and labels.
  • Automatic graph partitioning.
  • Highly performant and scalable.

Project is in alpha version, there might be breaking changes in the future and they will be documented in the changelog.

Usage

Install pip package:

python -m pip install deepgnn

If you want to build package from source, see instructions in CONTRIBUTING.md.

Train and evaluate a graphsage model with pytorch on cora dataset:

cd examples/pytorch
python sage.py

Migrating Scripts

We provide a python module to help you upgrade your scripts to new deepgnn versions.

pip install google-pasta
python -m deepgnn.migrate.0_1_56 --script_dir directory_to_migrate

See CHANGELOG.md for full change details.

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distributions

No source distribution files available for this release.See tutorial on generating distribution archives.

Built Distributions

deepgnn_ge-0.1.62.dev1-py3-none-win_amd64.whl (3.1 MB view details)

Uploaded Python 3 Windows x86-64

deepgnn_ge-0.1.62.dev1-py3-none-macosx_10_9_x86_64.whl (4.6 MB view details)

Uploaded Python 3 macOS 10.9+ x86-64

File details

Details for the file deepgnn_ge-0.1.62.dev1-py3-none-win_amd64.whl.

File metadata

File hashes

Hashes for deepgnn_ge-0.1.62.dev1-py3-none-win_amd64.whl
Algorithm Hash digest
SHA256 1d427ab74f36dc5b56ec97bdc9a99a4b2c4878ed98bade623b9bd65591d858c1
MD5 335333a07d736e2f9f1c54ca2f41b028
BLAKE2b-256 9d48a8b8c90426d1377c062e1884e1d9e30aa5d18dd3d0358fbdfdd2995e9c0d

See more details on using hashes here.

File details

Details for the file deepgnn_ge-0.1.62.dev1-py3-none-manylinux1_x86_64.whl.

File metadata

File hashes

Hashes for deepgnn_ge-0.1.62.dev1-py3-none-manylinux1_x86_64.whl
Algorithm Hash digest
SHA256 c98fee79a21ffa0307f28da562c30ddf8f3f5f3c56e0c8bcb67d718a845fe120
MD5 4b11aa5465c7361f0aa55865c93c541d
BLAKE2b-256 fd808217fb35e86b66088f7245561d65f5f3ff62ae23b97d8a75cfc83a29c0bc

See more details on using hashes here.

File details

Details for the file deepgnn_ge-0.1.62.dev1-py3-none-macosx_10_9_x86_64.whl.

File metadata

File hashes

Hashes for deepgnn_ge-0.1.62.dev1-py3-none-macosx_10_9_x86_64.whl
Algorithm Hash digest
SHA256 b7dd9d4d24805109f4d64777e92d8b46099929bc4c4c7d5c469a945c960a6034
MD5 b12d3b9de6bae0024015afd0a971d70c
BLAKE2b-256 dd5eef1771f0b28d1724122942633d6698163c13490ec84e16d6cce9c778dfd4

See more details on using hashes here.

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page