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-py3-none-win_amd64.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 2aedc06094d6eac78f32eeb44d408531e7abb61e9066dfdadd7d7ef74b8e89b0 |
|
MD5 | 080e6c7b44a92378d875841710b60083 |
|
BLAKE2b-256 | 5421b64f5b2d7d1d1cb95c0dff144855276d7065c7288b27a0a3b4481b8cbda7 |
Hashes for deepgnn_ge-0.1.55-py3-none-manylinux1_x86_64.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 113e889dec62135c5b59cdefa54dd2beee9fbf65dc6948f57b75c47720917d59 |
|
MD5 | 99474bf2cb5f7538314e7b2e4e61e468 |
|
BLAKE2b-256 | 506f19924f7e10e587877e1d6b8c419db58a0fec383c7542dc6eb1d9af15f021 |
Hashes for deepgnn_ge-0.1.55-py3-none-macosx_10_9_x86_64.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | bad7641cd0ff82a14e5f8d5efa35dda4593f1072c39163b6cc0cbda86c74fb55 |
|
MD5 | 90649d8adb58982c8ba505251233b29f |
|
BLAKE2b-256 | 24d89f134fe5e90aba37333695dcddfd57708936e3ed4c23eb54f36ad590a3c3 |