Skip to main content

LossFunction

reductionml_core::loss_function

Trait LossFunction 

Source 
pub trait LossFunction: Send {
    // 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

Dyn Compatibility§

This trait is dyn compatible.

In older versions of Rust, dyn compatibility was called "object safety".

Implementors§