Enum RegressionLoss 

pub enum RegressionLoss {
    SquaredError,
    Huber(HuberLoss),
}

Regression loss variants.

Used by LinearRegression to select the loss function applied to each update.

Variants

SquaredError

0.5 * (y - y_hat)^2

Huber(HuberLoss)

Huber loss, robust to outliers.

Implementations

impl RegressionLoss

pub fn loss(&self, prediction: f64, target: f64) -> f64

Compute the loss value given a prediction and target.

pub fn gradient(&self, prediction: f64, target: f64) -> f64

Compute the derivative of the loss with respect to the prediction.

Trait Implementations

impl Clone for RegressionLoss

fn clone(&self) -> RegressionLoss

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for RegressionLoss

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl Default for RegressionLoss

fn default() -> RegressionLoss

Returns the “default value” for a type.