Trait Model

Source

pub trait Model {
Show 16 methods    // Required methods
    fn feedforward(&mut self, train_data: Array2D);
    fn backpropagate(&mut self, expected: Array2D);
    fn optimize(&mut self);
    fn batch_size(&self) -> usize;
    fn set_batch_size(&mut self, batch_size: usize);
    fn set_batch_size_for_tests(&mut self, batch_size: usize);
    fn layer(&self, id: usize) -> &Box<dyn AbstractLayer>;
    fn layers_count(&self) -> usize;
    fn last_layer(&self) -> &Box<dyn AbstractLayer>;
    fn optimizer(&self) -> &Box<dyn WithParams>;
    fn optimizer_mut(&mut self) -> &mut Box<dyn WithParams>;
    fn output_params(&self) -> CpuParams;
    fn model_type(&self) -> &str;
    fn save_state(&self, filepath: &str) -> Result<(), Box<dyn Error>>;
    fn load_state(&mut self, filepath: &str) -> Result<(), Box<dyn Error>>;

    // Provided method
    fn last_layer_metrics(&self) -> Option<&Metrics> { ... }
}