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> OptimAlgorithm<M> for RMSPropwhere
    M: Optimizable<Inputs = Matrix<f64>, Targets = Matrix<f64>>,

impl<M> OptimAlgorithm<M> for StochasticGDwhere
    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