Skip to main content

Graph convolutional layers

Project description

Travis Coverage PyPI

Graph convolutional layers.

Install

pip install keras-gcn

Usage

GraphConv

import keras
from keras_gru import GraphConv


DATA_DIM = 3

data_layer = keras.layers.Input(shape=(None, DATA_DIM))
edge_layer = keras.layers.Input(shape=(None, None))
conv_layer = GraphConv(
    units=32,
    step_num=1,
)([data_layer, edge_layer])

step_num is the maximum distance of two nodes that could be considered as neighbors. If step_num is greater than 1, then the inputs of edges must be 0-1 matrices.

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

keras-gcn-0.2.tar.gz (2.6 kB view hashes)

Uploaded Source

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page