tensorflow/gnn

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

What it solves

TensorFlow プラットフォーム上で Graph Neural Networks（GNN）を構築するためのスケーラブルなフレームワークを提供し、同種グラフと異種グラフ（ノードやエッジのタイプが複数あるグラフ）の両方に対応します。

How it works

このライブラリは、複雑なグラフスキーマを表現するための専用 GraphTensor 型を使用します。大規模データベースをトレーニングや推論に適したサブグラフに変換するグラフサンプラー、GNN アーキテクチャ設計のための事前構築済み Keras レイヤーとモデルのセット、そしてトレーニングプロセスをオーケストレーションする高レベル API を含んでいます。

Who it’s for

TensorFlow エコシステムを利用して GNN を構築・スケールさせたい機械学習エンジニアや研究者向けです。

Highlights

  • 複数のノード・エッジタイプを持つ異種グラフスキーマをサポート。
  • 巨大データセット処理のための統合グラフサンプリングツール。
  • すぐに使えるモデルと Keras レイヤーで迅速な GNN モデリングが可能。
  • 高レベルのトレーニングオーケストレーション API。