« Back to API reference

neuralbench.callbacks.RecordingLevelEval

class neuralbench.callbacks.RecordingLevelEval

Callback to evaluate average prediction over each recording (timeline).

on_test_batch_end(trainer: Trainer, pl_module: LightningModule, outputs, batch, batch_idx, dataloader_idx=0) → None

Called when the test batch ends.

on_test_epoch_end(trainer: Trainer, pl_module: LightningModule) → None

Called when the test epoch ends.

on_test_epoch_start(trainer: Trainer, pl_module: LightningModule) → None

Called when the test epoch begins.

setup(trainer: Trainer, pl_module: LightningModule, stage: str) → None

Called when fit, validate, test, predict, or tune begins.

« Back to API reference