Struct Trainer 

pub struct Trainer { /* private fields */ }

Model trainer for neural networks.

Implementations

impl Trainer

pub fn new( model: Sequential, optimizer: Box<dyn Optimizer>, lossfn: Box<dyn Loss>, ) -> Self

Create a new trainer.

pub fn add_callback(&mut self, callback: Box<dyn TrainingCallback>)

Add a callback to the trainer.

pub fn train( &mut self, train_loader: &mut DataLoader, numepochs: usize, val_loader: Option<&mut DataLoader>, ) -> CoreResult<()>

Train the model.

pub const fn train_metrics(&self) -> &Metrics

Get training metrics.

pub fn val_metrics(&self) -> Option<&Metrics>

Get validation metrics.