知识图谱业务落地技术推荐之图神经网络算法库&图计算框架汇总

Posted 2023-03-03 汀、

1.PyTorch Geometric：

https://pytorch-geometric.readthedocs.io/en/latest/notes/introduction.html

PyG是一个基于PyTorch的用于处理不规则数据（比如图）的库，或者说是一个用于在图等数据上快速实现表征学习的框架。它的运行速度很快，训练模型速度可以达到DGL（Deep Graph Library ）v0.2 的40倍（数据来自论文）。除了出色的运行速度外，PyG中也集成了很多论文中提出的方法（GCN,SGC,GAT,SAGE等等）和常用数据集。因此对于复现论文来说也是相当方便。由于速度和方便的优势，毫无疑问，PyG是当前最流行和广泛使用的GNN库

2.GLEM：微软

https://github.com/AndyJZhao/GLEM