tsmixer_params

Typed parameters of the tsmixer family; importable without Keras.

TsMixerBlockParams()

TsMixer block parameters

model_config = ConfigDict(frozen=True, extra='forbid')

norm: Literal['batch', 'layer'] = Field(default='layer', description='Normalization type')

Normalization type

activation: Literal['relu', 'gelu'] = Field(default='relu', description='Activation type')

Activation type

dropout: float = Field(default=None, description='Dropout rate')

Dropout rate

ff_dim: int = Field(default=None, description='Feed forward dimension')

Feed forward dimension

TsMixerParams()

TsMixer parameters

model_config = ConfigDict(frozen=True, extra='forbid')

family: Literal['tsmixer'] = 'tsmixer'

Model family

blocks: list[TsBlockParams] = Field(default_factory=list, description='UNext blocks')

TsMixer blocks

num_classes: int | None = Field(default=None, gt=0, description='Output steps; required')

Output steps; required