2023-12-18 05: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-18 09:15:29 +08:00
|
|
|
cosine_similarity_loss
|
2023-12-22 14:13:41 +08:00
|
|
|
cross_entropy
|
2024-01-18 09:15:29 +08:00
|
|
|
hinge_loss
|
|
|
|
|
huber_loss
|
2023-12-22 14:13:41 +08:00
|
|
|
kl_div_loss
|
2023-12-18 05:23:03 +08:00
|
|
|
l1_loss
|
2024-01-18 09:15:29 +08:00
|
|
|
log_cosh_loss
|
2023-12-18 05:23:03 +08:00
|
|
|
mse_loss
|
|
|
|
|
nll_loss
|
2023-12-22 14:13:41 +08:00
|
|
|
smooth_l1_loss
|
2024-01-18 09:15:29 +08:00
|
|
|
triplet_loss
|
