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.54-py3-none-win_amd64.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 03a35cc2df27e6c3982449ba74334be3373befdb1b539183382f279146e00d23 |
|
MD5 | 15d8f1d5ebddc6a342d8c6465aa3db1d |
|
BLAKE2b-256 | 6edf79ff02a73ebe76909beb5a90feefda3617a4f70f9384a2d097efcf9d51da |
Hashes for deepgnn_ge-0.1.54-py3-none-manylinux1_x86_64.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 1c058f2d2f96dd5055903e3625e2548e8fce84a9b0d70d7a6f2d82baf20e00a6 |
|
MD5 | cd94294208e41fbf108fac081f452e4e |
|
BLAKE2b-256 | 1d9bdb9179840c6183226890fad81fdcf1a5d456499624a95bad772dcb7d74d2 |
Hashes for deepgnn_ge-0.1.54-py3-none-macosx_10_9_x86_64.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | a32adbc8053b21109c868f4e69329aecec1ac45f3e983d86c764966ba0435f08 |
|
MD5 | a82f26909ee01b0fc7db84453b16f80a |
|
BLAKE2b-256 | f731d26cf2507f7130866b95538a52f307be6750b0e833ff40c11c3603f59bc0 |