Trait OptimAlgorithm 

pub trait OptimAlgorithm<M: Optimizable> {
    // Required method
    fn optimize(
        &self,
        model: &M,
        start: &[f64],
        inputs: &M::Inputs,
        targets: &M::Targets,
    ) -> Vec<f64>;
}

Trait for optimization algorithms.

Required Methods

fn optimize( &self, model: &M, start: &[f64], inputs: &M::Inputs, targets: &M::Targets, ) -> Vec<f64>

Return the optimized parameter using gradient optimization.

Takes in a set of starting parameters and related model data.

Implementors

impl<M> OptimAlgorithm<M> for RMSProp

where M: Optimizable<Inputs = Matrix<f64>, Targets = Matrix<f64>>,

impl<M> OptimAlgorithm<M> for StochasticGD

where M: Optimizable<Inputs = Matrix<f64>, Targets = Matrix<f64>>,

impl<M: Optimizable> OptimAlgorithm<M> for ConjugateGD

impl<M: Optimizable> OptimAlgorithm<M> for GradientDesc

impl<M: Optimizable<Inputs = Matrix<f64>, Targets = Matrix<f64>>> OptimAlgorithm<M> for AdaGrad