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。