L-electron-Rare/mlx
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add Hinge, Huber and LogCosh losses (#199)

Browse Source
This commit is contained in:
Nicholas Santavas
2023-12-22 10:28:10 -08:00
committed by GitHub
parent e8deca84e0
commit d35fa1db41
4 changed files with 117 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
docs/src/python/nn/losses.rst
+4 -1
View File
@@ -16,4 +16,7 @@ Loss Functions
mse_loss
nll_loss
smooth_l1_loss
triplet_loss
triplet_loss
hinge_loss
huber_loss
log_cosh_loss