nabla_ml::nab_loss

Trait Loss

Source 
pub trait Loss {
    // Required methods
    fn forward(&self, predictions: &NDArray, targets: &NDArray) -> f64;
    fn backward(&self, predictions: &NDArray, targets: &NDArray) -> NDArray;
}

Required Methods§

Source

fn forward(&self, predictions: &NDArray, targets: &NDArray) -> f64

Source

fn backward(&self, predictions: &NDArray, targets: &NDArray) -> NDArray

Implementors§