Struct Trainer 

pub struct Trainer<M: Module, O: Optimizer> {
    pub model: M,
    /* private fields */
}

Trainer de alto nivel

Ejemplo:

let trainer = Trainer::new(model, optimizer, MSELoss);
let metrics = trainer.train_epoch(&dataset)?;
println!("Loss: {:.4}", metrics.loss);

Fields

model: M

Implementations

impl<M: Module, O: Optimizer> Trainer<M, O>

pub fn new(model: M, optimizer: O, loss_fn: Box<dyn LossFunction>) -> Self

pub fn model(&self) -> &M

Get reference to the model (for debugging)

pub fn model_mut(&mut self) -> &mut M

Get mutable reference to the model

pub fn train_epoch(&mut self, dataset: &Dataset) -> Result<Metrics>

Entrena una época completa

pub fn evaluate(&self, dataset: &Dataset) -> Result<Metrics>

Evalúa el modelo sin actualizar pesos

pub fn fit( &mut self, train_dataset: &Dataset, val_dataset: Option<&Dataset>, epochs: usize, verbose: bool, ) -> Result<Vec<(Metrics, Option<Metrics>)>>

Training loop completo con múltiples epochs