msp.layers package

Module contents

The mps.layers module includes utility to create different types of tensorflow custom layers to define MSP model architecture.

class msp.layers.ContextEmbedding(*args, **kwargs)

Bases: keras.engine.base_layer.Layer

build(input_shape)

Create the state of the layer (weights)

call(inputs)

class msp.layers.GGCNLayer(*args, **kwargs)

Bases: keras.engine.base_layer.Layer

build(input_shape)

Create the state of the layer (weights)

call(inputs, training=None)

class msp.layers.MHALayer(*args, **kwargs)

Bases: keras.engine.base_layer.Layer

build(input_shape)

Create the state of the layer (weights)

call(inputs, training=None)