8000 mlpack-testing/loss_functions at master · iamshnoo/mlpack-testing · GitHub
[go: up one dir, main page]

Skip to content

Latest commit

 

History

History

loss_functions

Loss Functions

  • Cross Entropy Error (Binary Cross Entropy)
  • Earth Mover Distance
  • Hinge Embedding Loss
  • Huber Loss / Smooth L1 Loss
  • KL Divergence Loss
  • L1 Loss
  • Log CosH Loss
  • Mean Bias Error
  • Mean Squared Error
  • Mean Squared Logarithmic Error
  • NLL Loss
  • Margin Ranking Loss (can't be done without major modifications, see notebook)
  • Cosine Embedding Loss (kartikdutt18 said he will be doing this)

Linked Issue

#2444

Linked PR

#2495

Link to Blogs

What this folder contains

Jupyter Notebooks comparing PyTorch/numpy implementations of each of the above loss functions with mlpack/armadillo implementations along with explanations for the possible errors in the existing implementations.

How to run

To run the notebooks, you would need pytorch, numpy and armadillo. To avoid issues with linking the libraries correctly, it is better to run the notebook in Google Colab.

Google Colab links for notebooks

0