GitPlanet

Cheap and reliable Node.js hosting starts at $3/month, and $1/month static HTML hosting

Created with love in Canada, visit hostnodejs.com today

Feel like to post an Ad? Learn Details
All Projects → tensorflow → gnn

tensorflow / gnn

Licence: Apache-2.0 License
TensorFlow GNN is a library to build Graph Neural Networks on the TensorFlow platform.

Programming Languages

python
139335 projects - #7 most used programming language

Labels

tensorflowgnn

Projects that are alternatives of or similar to gnn

Awesome-Federated-Learning-on-Graph-and-GNN-papers
Federated learning on graph, especially on graph neural networks (GNNs), knowledge graph, and private GNN.
Stars: ✭ 206 (-63.08%)
Mutual labels:  gnn
gnn-re-ranking
A real-time GNN-based method. Understanding Image Retrieval Re-Ranking: A Graph Neural Network Perspective
Stars: ✭ 64 (-88.53%)
Mutual labels:  gnn
GNNs-in-Network-Neuroscience
A review of papers proposing novel GNN methods with application to brain connectivity published in 2017-2020.
Stars: ✭ 92 (-83.51%)
Mutual labels:  gnn
ncem
Learning cell communication from spatial graphs of cells
Stars: ✭ 77 (-86.2%)
Mutual labels:  gnn
BasicGNNTracking
This shows a basic implementation of the global nearest neighbour (GNN) multi target Tracker. Kalman filter is used for Tracking and Auction Algorithm for determining the assignment of measurments to filters.
Stars: ✭ 36 (-93.55%)
Mutual labels:  gnn
gnn-lspe
Source code for GNN-LSPE (Graph Neural Networks with Learnable Structural and Positional Representations), ICLR 2022
Stars: ✭ 165 (-70.43%)
Mutual labels:  gnn
awesome-efficient-gnn
Code and resources on scalable and efficient Graph Neural Networks
Stars: ✭ 498 (-10.75%)
Mutual labels:  gnn
Meta-Fine-Tuning
[CVPR 2020 VL3] The repository for meta fine-tuning in cross-domain few-shot learning.
Stars: ✭ 29 (-94.8%)
Mutual labels:  gnn
GNN-Recommender-Systems
An index of recommendation algorithms that are based on Graph Neural Networks.
Stars: ✭ 505 (-9.5%)
Mutual labels:  gnn
VectorNet
Pytorch implementation of CVPR2020 paper “VectorNet: Encoding HD Maps and Agent Dynamics from Vectorized Representation”
Stars: ✭ 88 (-84.23%)
Mutual labels:  gnn
mmgnn textvqa
A Pytorch implementation of CVPR 2020 paper: Multi-Modal Graph Neural Network for Joint Reasoning on Vision and Scene Text
Stars: ✭ 41 (-92.65%)
Mutual labels:  gnn
BGCN
A Tensorflow implementation of "Bayesian Graph Convolutional Neural Networks" (AAAI 2019).
Stars: ✭ 129 (-76.88%)
Mutual labels:  gnn
stagin
STAGIN: Spatio-Temporal Attention Graph Isomorphism Network
Stars: ✭ 34 (-93.91%)
Mutual labels:  gnn
Causing
Causing: CAUsal INterpretation using Graphs
Stars: ✭ 47 (-91.58%)
Mutual labels:  gnn
Literatures-on-GNN-Acceleration
A reading list for deep graph learning acceleration.
Stars: ✭ 50 (-91.04%)
Mutual labels:  gnn
spatio-temporal-brain
A Deep Graph Neural Network Architecture for Modelling Spatio-temporal Dynamics in rs-fMRI Data
Stars: ✭ 22 (-96.06%)
Mutual labels:  gnn
GCL
List of Publications in Graph Contrastive Learning
Stars: ✭ 25 (-95.52%)
Mutual labels:  gnn
mtad-gat-pytorch
PyTorch implementation of MTAD-GAT (Multivariate Time-Series Anomaly Detection via Graph Attention Networks) by Zhao et. al (2020, https://arxiv.org/abs/2009.02040).
Stars: ✭ 85 (-84.77%)
Mutual labels:  gnn
gemnet pytorch
GemNet model in PyTorch, as proposed in "GemNet: Universal Directional Graph Neural Networks for Molecules" (NeurIPS 2021)
Stars: ✭ 80 (-85.66%)
Mutual labels:  gnn
PDN
The official PyTorch implementation of "Pathfinder Discovery Networks for Neural Message Passing" (WebConf '21)
Stars: ✭ 44 (-92.11%)
Mutual labels:  gnn
View All Similar Projects ➔

TensorFlow GNN

This is an early (alpha) release to get community feedback. It's under active development and we may break API compatibility in the future.

TensorFlow GNN is a library to build Graph Neural Networks on the TensorFlow platform. It contains the following components:

  • A high-level Keras-style API to create GNN models that can easily be composed with other types of models. GNNs are often used in combination with ranking, deep-retrieval (dual-encoders) or mixed with other types of models (image, text, etc.)

  • GNN API for heterogeneous graphs. Many of the graph problems we approach at Google and in the real world contain different types of nodes and edges. Hence the emphasis in heterogeneous models.

  • A well-defined schema to declare the topology of a graph, and tools to validate it. It describes the shape of its training data and serves to guide other tools.

  • A GraphTensor composite tensor type which holds graph data, can be batched, and has efficient graph manipulation functionality available.

  • A library of operations on the GraphTensor structure:

    • Various efficient broadcast and pooling operations on nodes and edges, and related tools.

    • A library of standard baked convolutions, that can be easily extended by ML engineers/researchers.

    • A high-level API for product engineers to quickly build GNN models without necessarily worrying about its details.

  • A set of tools used to convert graph datasets and sample from large graphs.

  • An encoding of graph-shaped training data on file, as well as a library used to parse this data into a data structure your model can extract the various features.

This library is an OSS port of a Google internal library used in a broad variety of contexts, on homogeneous and heterogeneous graphs, and in conjunction with other scalable graph mining tools.

Installation Instructions

Installation from source.

This is currently the only way to install the preview release of tensorflow_gnn. A virtual environment is highly recommended.

  1. Clone tensorflow_gnn

    $> git clone https://github.com/tensorflow/gnn.git tensorflow_gnn

  2. Install TensorFlow

    TF-GNN currently uses tf.ExtensionTypes, which is a feature of TensorFlow 2.7. As such, you will need to install TensorFlow build, following the instructions here: https://www.tensorflow.org/install/pip.

    $> pip install tensorflow

  3. Install Bazel

    Bazel is required to build the source of this package. Follow the instructions here to install Bazel for your OS: https://docs.bazel.build/versions/main/install.html

  4. Install GraphViz

    This package uses GraphViz for visualization tools. Installation instructions vary depending on your operating system. E.g. for Ubuntu:

    $> sudo apt-get install graphviz graphviz-dev

  5. Install tensorflow_gnn

    $> cd tensorflow_gnn && python3 -m pip install .

Note that the project description data, including the texts, logos, images, and/or trademarks, for each open source project belongs to its rightful owner. If you wish to add or remove any projects, please contact us at [email protected].