Struct SGD 

pub struct SGD { /* private fields */ }

Implementations

impl SGD

pub fn new(learning_rate: f64, momentum: f64, decay: f64) -> SGD

pub fn default() -> SGD

Trait Implementations

impl Optimizer for SGD

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)

impl Summary for SGD

fn summerize(&self) -> String