losshub

Classification Loss Function Library.

Loss Functions for Image Classification

Rmse: $y = \sqrt{\frac{1}{n} \sum_{i=1}^{n} (y_i - y')^2}$

Mse: $y = \frac{1}{n} \sum_{i=1}^{n} (y_i - y')^2$

Installation

pip install losshub

Usage

from losshub.losses import mse, rmse
# outputs and labels
y_true = [1, 2, 3, 4, 5]
y_pred = [1, 2, 3, 4, 5]
# mse
mse(y_true, y_pred)
# rmse
rmse(y_true, y_pred)

References

• Balanced-Loss
• Rmse
• Mse