本文介绍中山大学图学习团队开发的图神经网络基准模型库GraphGallery，支持多种深度学习框架（PyTorch与TensorFlow）以及两种图神经网络开发后端（PyG与DGL），能够帮助你快速训练和测试图神经网络模型。

项目地址：

https://github.com/EdisonLeeeee/GraphGallery

GraphGallery架构图

GraphGallery的架构主要包括输入数据流，模型构建，以及训练测试pipeline，用于对目前现有的GNN模型进行快速搭建。GraphGallery目前实现了节点分类任务主流的图神经网络模型（如GCN，GAT等），以及部分节点嵌入模型（如DeepWalk，Node2Vec等）。

内容中包含的图片若涉及版权问题，请及时与我们联系删除