2023-12-17 13:20:55 -08:00
|
|
|
.. _losses:
|
|
|
|
|
|
|
|
|
|
.. currentmodule:: mlx.nn.losses
|
|
|
|
|
|
|
|
|
|
Loss Functions
|
|
|
|
|
--------------
|
|
|
|
|
|
|
|
|
|
.. autosummary::
|
|
|
|
|
:toctree: _autosummary_functions
|
|
|
|
|
:template: nn-module-template.rst
|
|
|
|
|
|
|
|
|
|
binary_cross_entropy
|
2024-01-17 11:37:50 +08:00
|
|
|
cosine_similarity_loss
|
2023-12-18 23:26:21 +08:00
|
|
|
cross_entropy
|
2024-01-18 22:44:44 +08:00
|
|
|
gaussian_nll_loss
|
2024-01-17 11:37:50 +08:00
|
|
|
hinge_loss
|
|
|
|
|
huber_loss
|
2023-12-18 23:26:21 +08:00
|
|
|
kl_div_loss
|
2023-12-17 13:20:55 -08:00
|
|
|
l1_loss
|
2024-01-17 11:37:50 +08:00
|
|
|
log_cosh_loss
|
2024-02-03 02:57:31 +08:00
|
|
|
margin_ranking_loss
|
2023-12-17 13:20:55 -08:00
|
|
|
mse_loss
|
|
|
|
|
nll_loss
|
2023-12-18 23:26:21 +08:00
|
|
|
smooth_l1_loss
|
2024-01-17 11:37:50 +08:00
|
|
|
triplet_loss
|
