图引擎 - 分布式图引擎,用于托管图。
项目描述
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
获取完整的变更详情。
项目详情
关闭
哈希值 for deepgnn_ge-0.1.64-py3-none-manylinux1_x86_64.whl
算法 | 哈希摘要 | |
---|---|---|
SHA256 | 02cc1888598a5dde99f4329a4b30eff2ab575acda4b58bc9a05f516742a3fb53 |
|
MD5 | c4e849b0554afda8d200d4969b1e6eb7 |
|
BLAKE2b-256 | 8d984d612134b4a589292f71f59529df9ac9d667f3ba0af198ebb4ed10434f60 |
关闭
哈希值 for deepgnn_ge-0.1.64-py3-none-macosx_10_9_x86_64.whl
算法 | 哈希摘要 | |
---|---|---|
SHA256 | 71427708bd4306977413212dd8f6c59a467a4a4de67bbc6214570e02f4355907 |
|
MD5 | f4d814d8c2cc85c58212cd8b569cb490 |
|
BLAKE2b-256 | 6f937c0ab258e86a87d2e89cddaaacfa9e4d7cb292293235bcddd399f003239c |