跳转到主要内容

图引擎 - 分布式图引擎,用于托管图。

项目描述

DeepGNN概述

DeepGNN是一个用于在大规模图数据上训练机器学习模型的框架。DeepGNN包含所有必要的功能,包括

  • 在CPU和GPU上分布式GNN训练和推理。
  • 自定义图神经网络设计。
  • 在线采样:图引擎(GE)将加载所有图数据,每个训练工作将调用GE以获取节点/边/邻居特征和标签。
  • 自动图分区。
  • 高性能且可扩展。

项目处于alpha版本,未来可能会有破坏性变更,并在变更日志中记录。

用法

安装pip包

python -m pip install deepgnn

如果您想从源构建包,请参阅CONTRIBUTING.md中的说明。

使用pytorch在cora数据集上训练和评估graphsage模型

cd examples/pytorch
python sage.py

迁移脚本

我们提供了一个Python模块,帮助您将脚本升级到新的deepgnn版本。

pip install google-pasta
python -m deepgnn.migrate.0_1_56 --script_dir directory_to_migrate

请参阅CHANGELOG.md获取完整的变更详情。

项目详情


下载文件

下载适用于您平台的文件。如果您不确定选择哪个,请了解更多关于安装包的信息。

源代码分发

此版本没有可用的源代码分发文件。请参阅生成分发存档的教程

构建分发

deepgnn_ge-0.1.64-py3-none-win_amd64.whl (5.7 MB 查看哈希值)

上传日期 Python 3 Windows x86-64

deepgnn_ge-0.1.64-py3-none-manylinux1_x86_64.whl (11.9 MB 查看哈希值)

上传日期 Python 3

deepgnn_ge-0.1.64-py3-none-macosx_10_9_x86_64.whl (4.6 MB 查看哈希值)

上传日期 Python 3 macOS 10.9+ x86-64

支持者