Trait rusty_machine::learning::optim::Optimizable

pub trait Optimizable {
    type Data;
    type Target;
    fn compute_grad(&self, params: &[f64], data: &Self::Data, target: &Self::Target) -> Vec<f64>;
}

Trait for models which can be gradient-optimized.

Associated Types

type Data

type Target

Required Methods

fn compute_grad(&self, params: &[f64], data: &Self::Data, target: &Self::Target) -> Vec<f64>

Compute the gradient for the model.

Implementors

• impl<'a> Optimizable for NeuralNet<'a>