Struct CallbackList 

pub struct CallbackList { /* private fields */ }

List of callbacks to execute in order.

Implementations

impl CallbackList

pub fn new() -> Self

Create a new callback list.

pub fn add(&mut self, callback: Box<dyn Callback>)

Add a callback to the list.

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

Execute on_train_begin for all callbacks.

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

Execute on_train_end for all callbacks.

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

Execute on_epoch_begin for all callbacks.

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

Execute on_epoch_end for all callbacks.

pub fn on_batch_begin( &mut self, batch: usize, state: &TrainingState, ) -> TrainResult<()>

Execute on_batch_begin for all callbacks.

pub fn on_batch_end( &mut self, batch: usize, state: &TrainingState, ) -> TrainResult<()>

Execute on_batch_end for all callbacks.

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

Execute on_validation_end for all callbacks.

pub fn should_stop(&self) -> bool

Check if any callback requests early stopping.

Trait Implementations

impl Default for CallbackList

fn default() -> Self

Returns the “default value” for a type.