← Back to API reference

neuraltrain.models.fmri_mlp.FmriLinear

pydantic model neuraltrain.models.fmri_mlp.FmriLinear

Single linear layer for fMRI decoding with temporal aggregation.

Parameters:

• out_dim (int or None) – Explicit output dimension. Overridden by n_outputs at build time.

• time_agg ({"in_mean", "in_linear", "out_mean", "out_linear"}) – Temporal aggregation strategy. "in_*" aggregates before the linear layer; "out_*" aggregates after.

• output_head_config (Mlp, dict[str, Mlp], or None) – Optional output head(s) appended after the linear layer.

Fields:

• out_dim (int | None)

• output_head_config (neuraltrain.models.common.Mlp | dict[str, neuraltrain.models.common.Mlp] | None)

• time_agg (Literal['in_mean', 'in_linear', 'out_mean', 'out_linear'])

field out_dim: int | None = None

field time_agg: Literal['in_mean', 'in_linear', 'out_mean', 'out_linear'] = 'out_linear'

field output_head_config: Mlp | dict[str, Mlp] | None = None

build(n_in_channels: int, n_outputs: int | None) → Module

← Back to API reference