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 타입을 사용합니다. 대규모 데이터베이스를 학습 및 추론에 적합한 서브그래프로 변환하는 그래프 샘플러, GNN 아키텍처 설계를 위한 사전 구축된 Keras 레이어와 모델 세트, 그리고 학습 과정을 조율하는 고수준 API를 포함합니다.

Who it’s for

TensorFlow 생태계에서 GNN을 구축하고 확장해야 하는 머신러닝 엔지니어 및 연구자를 위한 것입니다.

Highlights

  • 여러 노드 및 엣지 유형을 갖는 이종 그래프 스키마 지원.
  • 방대한 데이터셋 처리를 위한 통합 그래프 샘플링 도구.
  • 빠른 GNN 모델링을 위한 즉시 사용 가능한 모델 및 Keras 레이어.
  • 고수준 학습 오케스트레이션 API.