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.
Release files for deepgnn-ge 0.1.66
For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.
Built distributions (wheels)
| File | Interpreter | ABI | Platform | Reset |
|---|---|---|---|---|
| deepgnn_ge-0.1.66-py3-none-win_amd64.whl | Python 3 | none | Windows x86-64 | Details |
| deepgnn_ge-0.1.66-py3-none-manylinux1_x86_64.whl | Python 3 | none | Linux glibc 2.5+ x86-64 | Details |
Total release size: 15.2 MB
Release files / deepgnn_ge-0.1.66-py3-none-win_amd64.whl
| Download URL | deepgnn_ge-0.1.66-py3-none-win_amd64.whl |
|---|---|
| Size | 3.3 MB |
| Tags | Python 3 Windows x86-64 |
|
SHA-256 checksum How to use checksums |
bc4fc1d207c17d0245b78f670d7812a431d237448c675423ab383a62a216127d
|
|
BLAKE2b-256 checksum How to use checksums |
7e03eb1b72e9b6c46588a9dfcf0e43177832630250589c816fc4446e24506560
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
twine/5.1.1 CPython/3.11.9
|
Release files / deepgnn_ge-0.1.66-py3-none-manylinux1_x86_64.whl
| Download URL | deepgnn_ge-0.1.66-py3-none-manylinux1_x86_64.whl |
|---|---|
| Size | 11.9 MB |
| Tags | Linux glibc 2.5+ x86-64 Python 3 |
|
SHA-256 checksum How to use checksums |
5eca280ccd090473706a6c1107828f59263dc25d8726c05fd1476f0f25f4f851
|
|
BLAKE2b-256 checksum How to use checksums |
62907e3ed1893d8d39b18c728be2ded7e0bbbc47e96ddd7f6792719f1de0ce07
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
twine/5.1.1 CPython/3.11.9
|
