Trait rusty_machine::learning::optim::OptimAlgorithm

pub trait OptimAlgorithm<M: Optimizable> {
    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: Optimizable> OptimAlgorithm<M> for GradientDesc
• impl<M> OptimAlgorithm<M> for StochasticGD where M: Optimizable<Inputs=Matrix<f64>, Targets=Matrix<f64>>
• impl<M: Optimizable<Inputs=Matrix<f64>, Targets=Matrix<f64>>> OptimAlgorithm<M> for AdaGrad
• impl<M: Optimizable> OptimAlgorithm<M> for ConjugateGD