ts2.models.single_session

Baselines fit on one session, from statistical to sequence models.

AutoencoderMLP	Simple Autoencoder with MLP encoder/decoder for TS2 neural prediction.
LFADS	LFADS for TS2 co-smoothing and forecasting [Keshtkaran et al., 2022, Pandarinath et al., 2018].
LFADSEvalTrainer	TS2 trainer for LFADS.
NDT	Single-session Transformer encoder over binned spikes [Ye and Pandarinath, 2021].
NDTEvalTrainer	TS2 trainer for the single-session NDT.

Statistical baselines

StatBaseline	Shared skeleton for every statistical baseline.
PopCoupling	Population coupling.
RRRReadout	Reduced-rank readout over binned counts.
RRRReadoutWithISI	Reduced-rank readout over [counts, time-since-last-spike, last ISI].
TrailingMean	Trailing-window mean, held flat over the horizon.
Shrinkage	Trial-local / global rate blend, held flat over the horizon.
RidgeAR	Per-unit ridge autoregression from the observed bins to the horizon.
StatBaselineTrainer	TS2 trainer for the zero-parameter statistical baselines.