Trait reductionml_core::loss_function::LossFunction

source · 
pub trait LossFunction {
    // Required methods
    fn get_type(&self) -> LossFunctionType;
    fn get_loss(
        &self,
        min_label: f32,
        max_label: f32,
        prediction: f32,
        label: f32
    ) -> f32;
    fn get_update(
        &self,
        prediction: f32,
        label: f32,
        update_scale: f32,
        pred_per_update: f32
    ) -> f32;
    fn get_unsafe_update(
        &self,
        prediction: f32,
        label: f32,
        update_scale: f32
    ) -> f32;
    fn get_square_grad(&self, prediction: f32, label: f32) -> f32;
    fn first_derivative(
        &self,
        min_label: f32,
        max_label: f32,
        prediction: f32,
        label: f32
    ) -> f32;
    fn second_derivative(
        &self,
        min_label: f32,
        max_label: f32,
        prediction: f32,
        label: f32
    ) -> f32;
}

Required Methods§

source

fn get_type(&self) -> LossFunctionType

source

fn get_loss( &self, min_label: f32, max_label: f32, prediction: f32, label: f32 ) -> f32

source

fn get_update( &self, prediction: f32, label: f32, update_scale: f32, pred_per_update: f32 ) -> f32

source

fn get_unsafe_update( &self, prediction: f32, label: f32, update_scale: f32 ) -> f32

source

fn get_square_grad(&self, prediction: f32, label: f32) -> f32

source

fn first_derivative( &self, min_label: f32, max_label: f32, prediction: f32, label: f32 ) -> f32

source

fn second_derivative( &self, min_label: f32, max_label: f32, prediction: f32, label: f32 ) -> f32

Implementors§