Google выпустила TensorFlow-GN — новую библиотеку для построения графовых нейронных сетей на платформе Tensorflow.

Исходный код доступен по лицензии Apache 2.0.

Ключевые моменты модели:

✅TF-GNN 1.0: API Python для динамического/пакетного ввода типа TensorraphTensor для представления графиков;

✅TF-GNN: построение/обучение GNN на различных абстракциях;

✅неоднородная схема: несколько типов узлов/ребер;

✅инструменты подготовки данных (например, DB->graph);

✅преобразование огромной базы данных в подграф разумного размера;

✅коллекция готовых к использованию моделей и слоев Keras;

✅API высокого уровня для обучения оркестрации.

Источник