Trait rusty_machine::learning::optim::Optimizable

pub trait Optimizable {
    type Inputs;
    type Targets;
    fn compute_grad(&self, params: &[f64], inputs: &Self::Inputs, targets: &Self::Targets) -> (f64, Vec<f64>);
}

Trait for models which can be gradient-optimized.

Associated Types

type Inputs

The input data type to the model.

type Targets

The target data type to the model.

Required Methods

fn compute_grad(&self, params: &[f64], inputs: &Self::Inputs, targets: &Self::Targets) -> (f64, Vec<f64>)

Compute the gradient for the model.

Implementors

• impl Optimizable for LinRegressor
• impl Optimizable for BaseLogisticRegressor
• impl<'a, T: Criterion> Optimizable for BaseNeuralNet<'a, T>