Trainer
encoding.trainer.AbstractTrainer
orchestrates the full pipeline:
Extract and downsample features per story
Apply FIR delays
Structure data (concat or train/test split)
Fit a model and log/save results
Configuration
Key constructor arguments:
assembly
: provides stories, brain/ timing arraysfeature_extractors
: list of extractors from the factorydownsampler
: aligns to TRsmodel
: must implementfit_predict
fir_delays
: list of sample delaystrimming_config
: indices for trimminguse_train_test_split
: bool, Lebel vs concatenateddataset_type
: e.g.,narratives
, used for caching keyslogger_backend
:wandb
ortensorboard
Outputs
Returns a metrics dict with common fields: median_score
, mean_score
, std_score
and optionally correlations
, significant_mask
.