Module loss 

Loss functions for training machine learning models.

Usage

use aprender::loss::{mse_loss, mae_loss, huber_loss};
use aprender::primitives::Vector;

let y_true = Vector::from_slice(&[1.0, 2.0, 3.0]);
let y_pred = Vector::from_slice(&[1.1, 2.1, 2.9]);

let mse = mse_loss(&y_pred, &y_true);
let mae = mae_loss(&y_pred, &y_true);
let huber = huber_loss(&y_pred, &y_true, 1.0);

Structs

HuberLoss

Huber loss function (struct wrapper).

MAELoss

Mean Absolute Error loss function (struct wrapper).

MSELoss

Mean Squared Error loss function (struct wrapper).

Traits

Loss

Trait for loss functions.

Functions

huber_loss

Huber loss (smooth approximation of MAE).

mae_loss

Mean Absolute Error (MAE) loss.

mse_loss

Mean Squared Error (MSE) loss.