Skip to main content

ModelTrainer

Trait ModelTrainer 

Source
pub trait ModelTrainer {
    // Required methods
    fn train(
        &self,
        examples: &[TrainingExample],
        config: &TrainingConfig,
    ) -> Result<Box<dyn TrainedModel>, TrainingError>;
    fn cross_validate(
        &self,
        examples: &[TrainingExample],
        config: &TrainingConfig,
    ) -> Result<CrossValidationResults, TrainingError>;
}
Expand description

Model trainer trait

Required Methods§

Source

fn train( &self, examples: &[TrainingExample], config: &TrainingConfig, ) -> Result<Box<dyn TrainedModel>, TrainingError>

Train model on examples

§Errors

Returns error if training fails or insufficient data

Source

fn cross_validate( &self, examples: &[TrainingExample], config: &TrainingConfig, ) -> Result<CrossValidationResults, TrainingError>

Run cross-validation

§Errors

Returns error if cross-validation fails

Implementors§