tsmixer

TsMixer Model

Overview

TsMixer is a fully MLP-based architecture for time series data.

For more info, refer to the original paper TsMixer: An All-MLP Architecture for Time Series.

Parameters are in helia_edge.models.tsmixer_params.

Functions

Name	Description
build	TsMixer model from TsMixerParams
ts_block	Residual block of TsMixer
norm_layer	Normalization layer
tsmixer_layer	TsMixer layer

Machine-readable model

• norm_layerfunctionNormalization layer
• ts_blockfunctionResidual block of TSMixer.
• tsmixer_layerfunctionTsMixer layer
• buildfunctionBuild a TsMixer model.

function

norm_layer

Python

Normalization layer

helia_edge/models/tsmixer.py:25

norm_layer(norm: str, name: str) -> keras.Layer

Normalization layer

Parameters

Parameters of norm_layer

Name	Type	Default	Description
norm	str	Required	Normalization type
name	str	Required	Name

Returns

Returns of norm_layer

Type	Description
keras.Layer	keras.Layer: Layer

function

ts_block

Python

Residual block of TSMixer.

helia_edge/models/tsmixer.py:54

ts_block(params: TsMixerBlockParams, name: str) -> keras.Layer

Residual block of TSMixer.

Parameters

Parameters of ts_block

Name	Type	Default	Description
params	TsBlockParams	Required	Block parameters
name	str	Required	Name

Returns

Returns of ts_block

Type	Description
keras.Layer	keras.Layer: Layer

function

tsmixer_layer

Python

TsMixer layer

helia_edge/models/tsmixer.py:93

tsmixer_layer(inputs: keras.KerasTensor, params: any) -> keras.KerasTensor

TsMixer layer

Parameters

Parameters of tsmixer_layer

Name	Type	Default	Description
inputs	keras.KerasTensor	Required	Input tensor
params	any	Required	Model parameters

Returns

Returns of tsmixer_layer

Type	Description
keras.KerasTensor	keras.KerasTensor: Output tensor

function

build

Python

Build a TsMixer model.

helia_edge/models/tsmixer.py:117

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

Build a TsMixer model.

Parameters

Parameters of build

Name	Type	Default	Description
params	TsMixerParams	Required	Model parameters.
input_shape	tuple[int | None, ...]	Required	Input shape without the batch axis; None for a variable axis.
batch_size	int | None	None	Static batch size; None for a dynamic batch.
name	str | None	None	Model name; the family when None.

Returns

Returns of build

Type	Description
keras.Model	keras.Model: The model, named ``tsmixer`` unless ``name`` is given.