tensorflow/gnn

TensorFlow GNN is a library to build Graph Neural Networks on the TensorFlow platform.

What it solves

它提供了一个可扩展的框架，用于在 TensorFlow 平台上构建图神经网络（GNN），特别支持同质图和异质图（具有多种节点和边类型的图）。

How it works

该库使用专门的 GraphTensor 类型来表示复杂的图结构。它包含图采样器，可将庞大的数据库转换为可管理的子图用于训练和推理，提供一套预构建的 Keras 层和模型用于 GNN 架构设计，并提供高级 API 来协调训练过程。

Who it’s for

需要在 TensorFlow 生态系统中构建和扩展 GNN 的机器学习工程师和研究人员。

Highlights

  • 支持具有多种节点和边类型的异构图模式。
  • 集成图采样工具以处理海量数据集。
  • 可直接使用的模型和 Keras 层，快速进行 GNN 建模。
  • 高级训练编排 API。