Trait neuro::optimizers::Optimizer

pub trait Optimizer {
    fn name(&self) -> &str;
    fn update_parameters(&mut self, layer: &mut dyn Layer, layer_idx: usize);
    fn initialize_parameters(&mut self, layers_dims: Vec<(Dim, Dim)>);
    fn save(&self, file: &File) -> Result<(), Error>;

    fn update_time_step(&mut self) { ... }
}

Defines the trait that needs to be implemented by any optimizer working with neuro.

Required methods

fn name(&self) -> &str

fn update_parameters(&mut self, layer: &mut dyn Layer, layer_idx: usize)

fn initialize_parameters(&mut self, layers_dims: Vec<(Dim, Dim)>)

fn save(&self, file: &File) -> Result<(), Error>

Loading content...

Provided methods

fn update_time_step(&mut self)

Loading content...

Implementors

impl Optimizer for AdaDelta

fn name(&self) -> &str

fn update_parameters(&mut self, layer: &mut dyn Layer, layer_idx: usize)

fn initialize_parameters(&mut self, layers_dims: Vec<(Dim, Dim)>)

fn save(&self, file: &File) -> Result<(), Error>

impl Optimizer for Adam

fn name(&self) -> &str

fn update_parameters(&mut self, layer: &mut dyn Layer, layer_idx: usize)

fn update_time_step(&mut self)

fn initialize_parameters(&mut self, layers_dims: Vec<(Dim, Dim)>)

fn save(&self, file: &File) -> Result<(), Error>

impl Optimizer for RMSProp

fn name(&self) -> &str

fn update_parameters(&mut self, layer: &mut dyn Layer, layer_idx: usize)

fn initialize_parameters(&mut self, layers_dims: Vec<(Dim, Dim)>)

fn save(&self, file: &File) -> Result<(), Error>

impl Optimizer for SGD

fn name(&self) -> &str

fn update_parameters(&mut self, layer: &mut dyn Layer, layer_idx: usize)

fn initialize_parameters(&mut self, layers_dims: Vec<(Dim, Dim)>)

fn save(&self, file: &File) -> Result<(), Error>

Loading content...