#182 · Primary category: Deep Learning Frameworks
spektral
Graph Neural Networks with Keras and Tensorflow 2.
Project last updated:01/21/24
GitHub Stars
2.4K
Forks
343
Contributors
27
License
MIT
Why we included this project
Graphs describe a lot of real-world data that tabular and sequence models handle awkwardly: social connections, molecular structures, and the links between documents or users. Spektral is a Python library built on the Keras API and TensorFlow 2 that brings graph neural networks to that stack, so teams already comfortable with Keras can train GNNs without adopting a separate framework. It bundles a broad set of established layers, including GCN, GraphSAGE, GAT, and GIN, plus pooling layers and utilities for transforming graph data into the tensors Keras expects. That covers common jobs like node classification, link prediction, and predicting molecular properties, all inside the standard Keras model-and-fit flow. The documentation and example templates make it easy to get a first model running and to see how graph inputs should be structured before you design your own.
Articles for this project
No articles for this project yet.
To suggest a topic or contribute an article, contact us.
Related projects in this category
tensorflow
An Open Source Machine Learning Framework for Everyone
pytorch
Tensors and Dynamic neural networks in Python with strong GPU acceleration
keras
Deep Learning for humans
nanoGPT
The simplest, fastest repository for training/finetuning medium-sized GPTs.
ray
Ray is an AI compute engine. Ray consists of a core distributed runtime and a set of AI Libraries for accelerating ML workloads.