tcn_params

Typed parameters of the TCN family; importable without Keras.

TcnBlockParamsclassTCN block parameters
TcnParamsclassTCN parameters
compact_tcn_paramsfunctionFour small SE4 blocks with 1/2/4/8 dilations and per-point linear output.

TcnBlockParams()

TCN block parameters

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

depth: int = Field(default=1, gt=0, description='Layer depth')

Layer depth

branch: int = Field(default=1, gt=0, description='Number of branches')

Number of branches

filters: int = Field(..., gt=0, description='# filters')

Number of filters

kernel: int | tuple[int, int] = Field(default=3, description='Kernel size')

Kernel size

dilation: int | tuple[int, int] = Field(default=1, description='Dilation rate')

Dilation rate

ex_ratio: float = Field(default=1, gt=0, allow_inf_nan=False, description='Expansion ratio')

Expansion ratio

se_ratio: float = Field(default=0, ge=0, allow_inf_nan=False, description='Squeeze and excite ratio')

Squeeze and excite ratio

dropout: float | None = Field(default=None, ge=0, lt=1, allow_inf_nan=False, description='Dropout rate')

Dropout rate

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

Normalization type

activation: str = Field(default='relu6', description='Activation function')

Activation function

TcnParams()

TCN parameters

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

family: Literal['tcn'] = 'tcn'

Model family

input_kernel: int | tuple[int, int] | None = Field(default=None, description='Input kernel size')

Input kernel size

input_norm: Literal['batch', 'layer'] | None = Field(default='layer', description='Input normalization type')

Input normalization type

block_type: Literal['lg', 'mb', 'sm'] = Field(default='mb', description='Block type')

Block type

blocks: list[TcnBlockParams] = Field(default_factory=list, description='TCN blocks')

TCN blocks

output_kernel: int | tuple[int, int] = Field(default=3, description='Output kernel size')

Output kernel size

include_top: bool = Field(default=True, description='Include top')

Include top

num_classes: int | None = Field(default=None, gt=0, description='Classes of the output layer; required with include_top')

Classes of the output layer; required with include_top

use_logits: bool = Field(default=True, description='Use logits')

Use logits

output_activation: str | None = Field(default=None, description='Output activation')

Output activation

compact_tcn_params(*, filters: int = 8, num_classes: int | None = None) -> TcnParams

Four small SE4 blocks with 1/2/4/8 dilations and per-point linear output.

The input shape is supplied to build by the caller. At least eight channels retain the builder’s SE squeeze path.