Trait rusty_machine::learning::optim::OptimAlgorithm
[−]
[src]
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>
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> OptimAlgorithm<M> for RMSProp where M: Optimizable<Inputs=Matrix<f64>, Targets=Matrix<f64>>
impl<M: Optimizable> OptimAlgorithm<M> for ConjugateGD