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.64-py3-none-win_amd64.whl (5.7 MB view details)

Uploaded Python 3 Windows x86-64

deepgnn_ge-0.1.64-py3-none-manylinux1_x86_64.whl (11.9 MB view details)

Uploaded Python 3

deepgnn_ge-0.1.64-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.64-py3-none-win_amd64.whl.

File metadata

File hashes

Hashes for deepgnn_ge-0.1.64-py3-none-win_amd64.whl
Algorithm Hash digest
SHA256 33bc22b0575e6acd26e782218658289810f6167410e1c709041871fa704cd3e0
MD5 dd5fa715414d5b29aa3c712edfadc411
BLAKE2b-256 d6a7619a83abd8820916d94266bab69478c7c07fdd532913c92e144b7efcb7f9

See more details on using hashes here.

File details

Details for the file deepgnn_ge-0.1.64-py3-none-manylinux1_x86_64.whl.

File metadata

File hashes

Hashes for deepgnn_ge-0.1.64-py3-none-manylinux1_x86_64.whl
Algorithm Hash digest
SHA256 02cc1888598a5dde99f4329a4b30eff2ab575acda4b58bc9a05f516742a3fb53
MD5 c4e849b0554afda8d200d4969b1e6eb7
BLAKE2b-256 8d984d612134b4a589292f71f59529df9ac9d667f3ba0af198ebb4ed10434f60

See more details on using hashes here.

File details

Details for the file deepgnn_ge-0.1.64-py3-none-macosx_10_9_x86_64.whl.

File metadata

File hashes

Hashes for deepgnn_ge-0.1.64-py3-none-macosx_10_9_x86_64.whl
Algorithm Hash digest
SHA256 71427708bd4306977413212dd8f6c59a467a4a4de67bbc6214570e02f4355907
MD5 f4d814d8c2cc85c58212cd8b569cb490
BLAKE2b-256 6f937c0ab258e86a87d2e89cddaaacfa9e4d7cb292293235bcddd399f003239c

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