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-torch
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/graphsage
./run.sh
Training other models
Examples folder contains various models one can experiment with DeepGNN. To train models with Tensorflow you need to install deepgnn-tf
package, deepgnn-torch
package contains packages to train pytorch examples. Each model folder contains a shell script run.sh
that will train a corresponding model on a toy graph, a README.md
file with a short description of a model, reference to original paper, and explanation of command line arguments.
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distributions
Built Distributions
Hashes for deepgnn_ge-0.1.55.dev1-py3-none-win_amd64.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 4857de7d9cbaf0183dc325b94ecd50e6fa2cf28267f12231ca5001be38fe1e89 |
|
MD5 | 7092dda6121f923ebe24fa4a76a9bd98 |
|
BLAKE2b-256 | 053504f6725a78ff9578346f4fb8a82cc195c377af8ba9c29df3104f811c3fbd |
Hashes for deepgnn_ge-0.1.55.dev1-py3-none-manylinux1_x86_64.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 2a36bbea51f375b7b98adf5d9ccae5da92282412d1bba292869e32632ed6b50b |
|
MD5 | 972cc859d601a4bb01f426da010cc1a1 |
|
BLAKE2b-256 | 176749aaf80556079a334afdce7a35bfe3f02d623eecd57162b23db9b08c8d2b |
Hashes for deepgnn_ge-0.1.55.dev1-py3-none-macosx_10_9_x86_64.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 241b36bc5e73816678b0765a78d56f239f2f7c521090b3343c04857b729916e0 |
|
MD5 | 9d12dd8aedd68aacfa8a746933da71c9 |
|
BLAKE2b-256 | fa6a2e17be6769e2d69b6ccc863e3f36a1520988019c0cec943b76293daf750d |