Module callback 

Callback system for training events

Provides extensible hooks for training loop events:

• on_train_begin / on_train_end
• on_epoch_begin / on_epoch_end
• on_step_begin / on_step_end

Example

use entrenar::train::callback::{TrainerCallback, CallbackContext, CallbackAction};

struct PrintCallback;

impl TrainerCallback for PrintCallback {
    fn on_epoch_end(&mut self, ctx: &CallbackContext) -> CallbackAction {
        println!("Epoch {} finished with loss {:.4}", ctx.epoch, ctx.loss);
        CallbackAction::Continue
    }
}

Structs

CallbackContext

Context passed to callbacks with current training state

CallbackManager

Manages multiple callbacks and dispatches events

CheckpointCallback

Checkpoint callback to save model state periodically

EarlyStopping

Early stopping callback to halt training when loss plateaus

ExplainabilityCallback

Callback for computing feature attributions during training

FeatureImportanceResult

Feature importance result for a single epoch

LRSchedulerCallback

Callback that applies a learning rate scheduler during training

MonitorCallback

Callback that integrates with entrenar’s monitoring system

ProgressCallback

Progress callback for logging training progress

Enums

CallbackAction

Action to take after a callback

ExplainMethod

Method for computing feature attributions

Traits

TrainerCallback

Trait for training callbacks