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) -> (f64, Vec<f64>);
}

Trait for models which can be gradient-optimized.

Associated Types

type Data

The input data type to the model.

type Target

The target data type to the model.

Required Methods

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

Compute the gradient for the model.

Implementors

• impl<'a, T: Criterion> Optimizable for NeuralNet<'a, T>