neuraltrain.models.linear.Linear¶

class neuraltrain.models.linear.Linear(*, reduction: Literal['mean', 'concat'] = 'mean', subject_layers_config: SubjectLayers | None = None)[source][source]¶

Simple linear projection, with optional per-subject weights.

Parameters:

reduction ({"mean", "concat"}) – How to reduce the time dimension before the linear layer. "mean" averages over time; "concat" flattens channels and time.
subject_layers_config (SubjectLayers or None) – If set, use a SubjectLayersModel instead of a shared nn.Linear.