https://github.com/microsoft/DeepGNN

DeepGNN is a framework for training machine learning models on large scale graph data.
https://github.com/microsoft/DeepGNN

Last synced: 2 months ago
JSON representation

DeepGNN is a framework for training machine learning models on large scale graph data.

• Host: GitHub
• URL: https://github.com/microsoft/DeepGNN
• Owner: microsoft
• License: mit
• Created: 2022-04-21T17:24:28.000Z (over 2 years ago)
• Default Branch: main
• Last Pushed: 2024-05-21T22:36:37.000Z (8 months ago)
• Last Synced: 2024-05-22T19:20:22.439Z (8 months ago)
• Language: Python
• Size: 6.19 MB
• Stars: 100
• Watchers: 9
• Forks: 26
• Open Issues: 3
• Metadata Files:
  • Readme: README.md
  • Changelog: CHANGELOG.md
  • Contributing: CONTRIBUTING.md
  • License: LICENSE
  • Code of conduct: CODE_OF_CONDUCT.md
  • Codeowners: .github/CODEOWNERS
  • Security: SECURITY.md
  • Support: SUPPORT.md

Awesome Lists containing this project

• awesome-gnn-systems - DeepGNN: a Framework for Training Machine Learning Models on Large Scale Graph Data

README

        
# 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:

```bash

python -m pip install deepgnn

```

If you want to build package from source, see instructions in [`CONTRIBUTING.md`](CONTRIBUTING.md).

Train and evaluate a graphsage model with pytorch on cora dataset:

```bash

cd examples/pytorch

python sage.py

```

## Migrating Scripts

We provide a python module to help you upgrade your scripts to new deepgnn versions.

```bash

pip install google-pasta

python -m deepgnn.migrate.0_1_56 --script_dir directory_to_migrate

```

See [`CHANGELOG.md`](CHANGELOG.md) for full change details.