2023-12-17 13:23:03 -08:00
|
|
|
.. _losses:
|
|
|
|
|
|
|
|
|
|
.. currentmodule:: mlx.nn.losses
|
|
|
|
|
|
|
|
|
|
Loss Functions
|
|
|
|
|
--------------
|
|
|
|
|
|
|
|
|
|
.. autosummary::
|
|
|
|
|
:toctree: _autosummary_functions
|
|
|
|
|
:template: nn-module-template.rst
|
|
|
|
|
|
|
|
|
|
binary_cross_entropy
|
2024-01-17 17:15:29 -08:00
|
|
|
cosine_similarity_loss
|
2023-12-21 22:13:41 -08:00
|
|
|
cross_entropy
|
2024-02-01 13:08:29 -08:00
|
|
|
gaussian_nll_loss
|
2024-01-17 17:15:29 -08:00
|
|
|
hinge_loss
|
|
|
|
|
huber_loss
|
2023-12-21 22:13:41 -08:00
|
|
|
kl_div_loss
|
2023-12-17 13:23:03 -08:00
|
|
|
l1_loss
|
2024-01-17 17:15:29 -08:00
|
|
|
log_cosh_loss
|
2024-02-08 12:44:23 -08:00
|
|
|
margin_ranking_loss
|
2023-12-17 13:23:03 -08:00
|
|
|
mse_loss
|
|
|
|
|
nll_loss
|
2023-12-21 22:13:41 -08:00
|
|
|
smooth_l1_loss
|
2024-01-17 17:15:29 -08:00
|
|
|
triplet_loss
|
