Skip to content
heliaEDGE
Reference
HELIA

composer

This module provides utility functions to compose a sequential set of networks/layers.

Parameters are in helia_edge.models.composer_params.

Functions

Name Description
build Composer model from ComposerParams
composer_layer Composes a sequential set of networks/layers

Machine-readable model

function

Composes a sequential set of networks/layers.

helia_edge/models/composer.py:28

composer_layer(x: keras.KerasTensor, params: ComposerParams) -> keras.KerasTensor

Composes a sequential set of networks/layers. Useful for adding custom layers to a pre-trained model (e.g. foundation).

Parameters of composer_layer
NameTypeDefaultDescription
xkeras.KerasTensorRequiredModel input
paramsComposerParamsRequiredModel parameters
Returns of composer_layer
TypeDescription
keras.KerasTensorkeras.KerasTensor: Model output
function

build

Python

Build a Composer model.

helia_edge/models/composer.py:73

build(
params: ComposerParams,
input_shape: tuple[int | None, ...],
*,
batch_size: int | None = None,
name: str | None = None,
) -> keras.Model

Build a Composer model.

Parameters of build
NameTypeDefaultDescription
paramsComposerParamsRequiredModel parameters.
input_shapetuple[int | None, ...]RequiredInput shape without the batch axis; None for a variable axis.
batch_sizeint | NoneNoneStatic batch size; None for a dynamic batch.
namestr | NoneNoneModel name; the family when None.
Returns of build
TypeDescription
keras.Modelkeras.Model: The model, named ``composer`` unless ``name`` is given.