zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-09-01 04:24:36 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

added mse_loss, nll_loss and kl_div_loss (#98)

Browse Source 
* added mse_loss, nll_loss and kl_div_loss

* fixed axis not defined error in nll_loss

* fixed axis not defined in kl_div_loss

* added tests for mse, nll and kl_div

* modified docstrings and added reduce helper func

* updated docstring in kl_div_loss and moved helper func

* added new kl divergence implementation

* added reduction to test

* updated docstring of kl_div_loss with correct spelling

* added losses to nn.rst in docs
This commit is contained in:
Enoch Kan
2023-12-09 22:25:03 +00:00
committed by GitHub
parent ac6dc5d3eb
commit 0b28399638
3 changed files with 142 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
docs/src/python/nn.rst
View File

@@ -180,3 +180,6 @@ Loss Functions
losses.cross_entropy
losses.l1_loss
losses.mse_loss
losses.nll_loss
losses.kl_div_loss