Classification Loss Function Library
Project description
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
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
losshub-0.0.6.tar.gz
(3.6 kB
view hashes)
