Trait renforce::util::GradientDescAlgo

pub trait GradientDescAlgo<F: Float> {
    fn calculate(&mut self, grad: Vec<F>, lr: F) -> Vec<F>;
}

Calculates gradient steps

Required Methods

fn calculate(&mut self, grad: Vec<F>, lr: F) -> Vec<F>

Calculates local step for minimizing function

Implementors

• impl<F: Float> GradientDescAlgo<F> for GradientDesc
• impl<F: Float> GradientDescAlgo<F> for GradDescMomentum<F>
• impl<F: Float> GradientDescAlgo<F> for RMSProp<F>