Struct Adam 

pub struct Adam { /* private fields */ }

Adaptive moment estimation gradient descent.

https://arxiv.org/pdf/1412.6980.pdf

Implementations

impl Adam

pub fn new(n_params: usize, learning_rate: f64, beta1: f64, beta2: f64) -> Adam

Trait Implementations

impl Clone for Adam

fn clone(&self) -> Adam

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for Adam

fn fmt(&self, f: &mut Formatter<'_>) -> Result<(), Error>

Formats the value using the given formatter.

impl Optimiser for Adam

fn step_scaled( &mut self, weights: &mut ArrayBase<ViewRepr<&mut f64>, Dim<[usize; 1]>>, features: &Features, scale_factor: f64, ) -> Result<(), Error>

fn reset(&mut self)

fn step( &mut self, weights: &mut ArrayBase<ViewRepr<&mut f64>, Dim<[usize; 1]>>, grad: &G, ) -> Result<(), Error>

impl PartialEq for Adam

fn eq(&self, other: &Adam) -> bool

Tests for self and other values to be equal, and is used by ==.

fn ne(&self, other: &Rhs) -> bool

Tests for !=. The default implementation is almost always sufficient, and should not be overridden without very good reason.

impl StructuralPartialEq for Adam