neural_network_rs::neural_network::optimizer

Trait Optimizer

Source 
pub trait Optimizer: Summary {
    // Required methods
    fn update_params(
        &mut self,
        layers: &mut Vec<Layer<'_>>,
        nabla_bs: &Vec<Array2<f64>>,
        nabla_ws: &Vec<Array2<f64>>,
    );
    fn initialize(&mut self, layers: &Vec<Layer<'_>>);
    fn pre_update(&mut self);
    fn post_update(&mut self);
}

Required Methods§

Source

fn update_params( &mut self, layers: &mut Vec<Layer<'_>>, nabla_bs: &Vec<Array2<f64>>, nabla_ws: &Vec<Array2<f64>>, )

Source

fn initialize(&mut self, layers: &Vec<Layer<'_>>)

Source

fn pre_update(&mut self)

Source

fn post_update(&mut self)

Implementors§