Skip to content

Latest commit

 

History

History
54 lines (40 loc) · 2.08 KB

README_cn.md

File metadata and controls

54 lines (40 loc) · 2.08 KB

GL pypi docs graph-learn CI License

简体中文 | English

Graph-Learn(原AliGraph) 是面向大规模图神经网络的研发和应用而设计的一款分布式框架。 它从大规模图训练的实际问题出发,提炼和抽象了一套适合于常见图神经网络模型的编程范式, 并已经成功应用在阿里巴巴内部的诸如搜索推荐、网络安全、知识图谱等众多场景。

Graph-Learn提供了图采样操作的Python和C++接口,并且提供了一个类似gremlin的GSL(Graph Sampling Language)接口。对于上层图学习模型,Graph-Learn提供了一套模型开发的范式和流程,兼容TensorFlow和PyTorch,提供了数据层,模型层接口和丰富的模型示例。

用户文档

安装部署

  1. 通过wheel包安装(linux, python3, glibc 2.24+)
pip install graph-learn
  1. 从源码编译

  2. 使用Docker

开始使用

GraphSAGE示例

cd examples/tf/ego_sage/
python train_unsupervised.py

分布式训练示例

论文

如果Graph-Learn对你的工作有所帮助,请引用如下论文。

@article{zhu2019aligraph,
  title={AliGraph: a comprehensive graph neural network platform},
  author={Zhu, Rong and Zhao, Kun and Yang, Hongxia and Lin, Wei and Zhou, Chang and Ai, Baole and Li, Yong and Zhou, Jingren},
  journal={Proceedings of the VLDB Endowment},
  volume={12},
  number={12},
  pages={2094--2105},
  year={2019},
  publisher={VLDB Endowment}
}

协议

Apache License 2.0.