Struct BestModelCheckpoint 

pub struct BestModelCheckpoint { /* private fields */ }

Saves model weights when monitored metric improves.

This struct tracks the best metric value and signals when a new best is found. The caller is responsible for performing the actual model serialisation.

Implementations

impl BestModelCheckpoint

pub fn new(save_path: PathBuf, mode: MonitorMode) -> Self

Create a new checkpoint tracker.

pub fn check(&mut self, value: f32) -> bool

Check if metric improved. Returns true if a new best was found.

pub fn save_path(&self) -> &Path

Get the save path.

pub fn best_value(&self) -> f32

Best value seen.

impl BestModelCheckpoint

pub fn with_monitor(self, key: impl Into<String>) -> Self

Set the metric key to monitor (default: "loss").

Trait Implementations

impl Clone for BestModelCheckpoint

fn clone(&self) -> BestModelCheckpoint

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for BestModelCheckpoint

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl TrainingCallback for BestModelCheckpoint

fn on_epoch_end( &mut self, _epoch: usize, metrics: &HashMap<String, f32>, ) -> bool

Called after each epoch. Returns true if training should stop.

fn on_batch_end(&mut self, _epoch: usize, _batch: usize, _loss: f32)

Called after each batch within an epoch. Default implementation does nothing.