Module: gcn

View source on GitHub

Graph Convolutional Networks.

Users of TF-GNN can use this model by importing it next to the core library as

import tensorflow_gnn as tfgnn
from tensorflow_gnn.models import gcn

Classes

class GCNConv: Implements the Graph Convolutional Network by Kipf&Welling (2016).

Functions

GCNHomGraphUpdate(...): Returns a graph update layer for GCN convolution.