Downloads · 30 days
5
1% of all-time downloads
keras-io/graph-attention-nets
graph-attention-nets is a machine learning model from keras-io. Use it for the machine learning task on the model card, and read the license before you ship it in a product. It is set up for tf-keras. The card lists the license as cc0-1.0.
This repo contains the model and the notebook to this Keras example on Graph Attention Networks for Node Classification.
Downloads · 30 days
5
1% of all-time downloads
All-time downloads
602
Public
Repo size
221 MB
Likes
6
Public
Click a slice to open those files.
.pb221 MB · 99%
From the Hugging Face model README
This repo contains the model and the notebook to this Keras example on Graph Attention Networks for Node Classification.
Full credits to: Alexander Kensert
Graph neural networks is the preferred neural network architecture for processing data structured as graphs (for example, social networks or molecule structures), yielding better results than fully-connected networks or convolutional networks.
This tutorial implements a specific graph neural network known as a Graph Attention Network (GAT) to predict labels of scientific papers based on the papers they cite (using the Cora dataset).
References For more information on GAT, see the original paper Graph Attention Networks as well as DGL's Graph Attention Networks documentation.