Skip to main content

Ctrl+K

IBL BWB

Dataset
Setup
Codebase
Pretraining
TS1: Behavior
TS2: Dynamics
TS3: Anatomy
Submission
API
Refs

Leaderboard
GitHub

Dataset
Setup
Codebase
Pretraining
TS1: Behavior
TS2: Dynamics
TS3: Anatomy
Submission
API
Refs

Leaderboard
GitHub

Section Navigation

All objects

Evaluation contract

ibl_bwb_eval
ibl_bwb_eval.metrics
ibl_bwb_eval.multi_unit
- multi_unit_prediction
ibl_bwb_eval.predictions
- PredictionsWriter
- version_of
ibl_bwb_eval.entity_ids
- encode_entity_ids
- decode_entity_ids

Scoring

ibl_bwb_eval.scoring.ts1
ibl_bwb_eval.scoring.ts2
ibl_bwb_eval.scoring.ts3
ibl_bwb_eval.scoring.aggregation
- from_ts3
- aggregate
- rank

Core

core.data
core.dataset
core.finetuning
core.model
- BaseModel
core.trainer
- BaseTrainer
core.samplers
- DistributedSamplerWrapper
- SessionBatchSampler
core.transforms
core.nn
core.nn.loss
- WeightedMSELoss
- CLIPLoss
core.nn.metrics
- rankme

Pretraining

pretrain.models
pretrain.datasets

TS1: Behavior Prediction

ts1
ts1.models.single_session
- Linear
- MLP
- GRU
- TCN
- CEBRA
- CEBRAEvalTrainer
- NDTSuperv
ts1.models.pretrained

TS2: Neural Activity Prediction

ts2
ts2.models.single_session
ts2.models.pretrained
- MtMEvalTrainer
- NDTStitchEvalTrainer

TS3: Brain Region Prediction

ts3
ts3.models.inductive
ts3.models.transductive
ts3.models.supervised
ts3.probes

API Reference
ts3.models.transductive
MtMCalibrateTrainer

MtMCalibrateTrainer#

class ts3.models.transductive.MtMCalibrateTrainer(cfg, rank, world_size)[source]#: Bases: pretrain.models.mtm.mtm_pretrain.MtMPretrain

previous

NDTStitchCalibrateTrainer

next

POSSMCalibrateTrainer

On this page

MtMCalibrateTrainer

© Copyright 2026, The IBL BrainWideBench authors.

Created using Sphinx 8.1.3.

Built with the PyData Sphinx Theme 0.19.0.