Trait alumina::opt::Optimiser

pub trait Optimiser<'a> {
    fn get_graph(&mut self) -> &mut Graph;
    fn add_boxed_step_callback(
        &mut self, 
        func: Box<FnMut(CallbackData) -> CallbackSignal>
    );
    fn optimise_from(
        &mut self, 
        training_set: &mut Supplier, 
        params: Vec<f32>
    ) -> Vec<f32>;
    fn step(
        &mut self, 
        training_set: &mut Supplier, 
        params: Vec<f32>
    ) -> (f32, Vec<f32>);

    fn add_step_callback<F: 'static + FnMut(CallbackData) -> CallbackSignal>(
        &mut self, 
        func: F
    ) { ... }
    fn optimise(&mut self, training_set: &mut Supplier) -> Vec<f32> { ... }
}

Required Methods

fn get_graph(&mut self) -> &mut Graph

fn add_boxed_step_callback(
    &mut self,
    func: Box<FnMut(CallbackData) -> CallbackSignal>
)

err, step, evaluations, graph, params -> returns whether the optimisation loop should continue

fn optimise_from(
    &mut self,
    training_set: &mut Supplier,
    params: Vec<f32>
) -> Vec<f32>

fn step(
    &mut self,
    training_set: &mut Supplier,
    params: Vec<f32>
) -> (f32, Vec<f32>)

Provided Methods

fn add_step_callback<F: 'static + FnMut(CallbackData) -> CallbackSignal>(
    &mut self,
    func: F
)

fn optimise(&mut self, training_set: &mut Supplier) -> Vec<f32>

Implementors

• impl<'a> Optimiser<'a> for Sgd<'a>
• impl<'a> Optimiser<'a> for Cain<'a>