Trait lfa::optim::Optimiser

pub trait Optimiser<G = Features> {
    fn step_scaled(
        &mut self, 
        weights: &mut ArrayViewMut1<f64>, 
        grad: &G, 
        scale_factor: f64
    ) -> Result<()>;

    fn step(&mut self, weights: &mut ArrayViewMut1<f64>, grad: &G) -> Result<()> { ... }
    fn reset(&mut self) { ... }
}

Required methods

fn step_scaled(
    &mut self,
    weights: &mut ArrayViewMut1<f64>,
    grad: &G,
    scale_factor: f64
) -> Result<()>

Loading content...

Provided methods

fn step(&mut self, weights: &mut ArrayViewMut1<f64>, grad: &G) -> Result<()>

fn reset(&mut self)

Loading content...

Implementors

impl Optimiser<Features> for AdaMax

fn step_scaled(
    &mut self,
    weights: &mut ArrayViewMut1<f64>,
    features: &Features,
    scale_factor: f64
) -> Result<()>

fn reset(&mut self)

impl Optimiser<Features> for Adagrad

fn step_scaled(
    &mut self,
    weights: &mut ArrayViewMut1<f64>,
    features: &Features,
    scale_factor: f64
) -> Result<()>

fn reset(&mut self)

impl Optimiser<Features> for Adam

fn step_scaled(
    &mut self,
    weights: &mut ArrayViewMut1<f64>,
    features: &Features,
    scale_factor: f64
) -> Result<()>

fn reset(&mut self)

impl Optimiser<Features> for ISGD

fn step(
    &mut self,
    weights: &mut ArrayViewMut1<f64>,
    features: &Features
) -> Result<()>

fn step_scaled(
    &mut self,
    weights: &mut ArrayViewMut1<f64>,
    features: &Features,
    scale_factor: f64
) -> Result<()>

impl Optimiser<Features> for NAG

fn step_scaled(
    &mut self,
    weights: &mut ArrayViewMut1<f64>,
    features: &Features,
    scale_factor: f64
) -> Result<()>

fn reset(&mut self)

impl Optimiser<Features> for SGD

fn step(
    &mut self,
    weights: &mut ArrayViewMut1<f64>,
    features: &Features
) -> Result<()>

fn step_scaled(
    &mut self,
    weights: &mut ArrayViewMut1<f64>,
    features: &Features,
    scale_factor: f64
) -> Result<()>

impl Optimiser<Features> for SGDMomentum

fn step_scaled(
    &mut self,
    weights: &mut ArrayViewMut1<f64>,
    features: &Features,
    scale_factor: f64
) -> Result<()>

fn reset(&mut self)

Loading content...