简体中文 | English
Graph-Learn(原AliGraph) 是面向大规模图神经网络的研发和应用而设计的一款分布式框架。 它从大规模图训练的实际问题出发,提炼和抽象了一套适合于常见图神经网络模型的编程范式, 并已经成功应用在阿里巴巴内部的诸如搜索推荐、网络安全、知识图谱等众多场景。
Graph-Learn提供了图采样操作的Python和C++接口,并且提供了一个类似gremlin的GSL(Graph Sampling Language)接口。对于上层图学习模型,Graph-Learn提供了一套模型开发的范式和流程,兼容TensorFlow和PyTorch,提供了数据层,模型层接口和丰富的模型示例。
- 通过wheel包安装(linux, python3, glibc 2.24+)
pip install graph-learn
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.