Struct EarlyStoppingCallback 

pub struct EarlyStoppingCallback {
    pub patience: usize,
    pub min_delta: f64,
    /* private fields */
}

Callback for early stopping based on validation loss.

Fields

patience: usize

Number of epochs with no improvement after which training will be stopped.

min_delta: f64

Minimum change to qualify as an improvement.

Implementations

impl EarlyStoppingCallback

pub fn new(patience: usize, min_delta: f64) -> Self

Create a new early stopping callback.

Trait Implementations

impl Callback for EarlyStoppingCallback

fn on_epoch_end( &mut self, epoch: usize, state: &TrainingState, ) -> TrainResult<()>

Called at the end of an epoch.

fn should_stop(&self) -> bool

Check if training should stop early.

fn on_train_begin(&mut self, _state: &TrainingState) -> TrainResult<()>

Called at the beginning of training.

fn on_train_end(&mut self, _state: &TrainingState) -> TrainResult<()>

Called at the end of training.

fn on_epoch_begin( &mut self, _epoch: usize, _state: &TrainingState, ) -> TrainResult<()>

Called at the beginning of an epoch.

fn on_batch_begin( &mut self, _batch: usize, _state: &TrainingState, ) -> TrainResult<()>

Called at the beginning of a batch.

fn on_batch_end( &mut self, _batch: usize, _state: &TrainingState, ) -> TrainResult<()>

Called at the end of a batch.

fn on_validation_end(&mut self, _state: &TrainingState) -> TrainResult<()>

Called after validation.