1. 主要库
库名 | 官方手册 | 论文及简介 | 源码 | 支持框架 |
---|---|---|---|---|
DGL :Deep Graph Library | 官方手册 | 论文 | 源码 | PyTorch, MXNet 和 TensorFlow |
PyG:pytorch_geometric | 官方手册 | 中文简介1、中文简介2 | 源码 | PyTorch |
TF-GNN | 官方手册 | 论文、中文简介、英文简介 | 源码 | TensorFlow 2 |
tf_geometric | 中文简介 | TensorFlow |
2. 其他
TF-GNN论文中“7 Related Work”中也提到了一些其他库,并对他们的特点进行了描述
TF Graph Neural Network Samples,由微软开发基于Tensorflow提供了多个现成网络的源码