https://github.com/unvercan/loss-function-comparison-pytorch

Comparison of common loss functions in PyTorch using MNIST dataset
https://github.com/unvercan/loss-function-comparison-pytorch

cross-entropy-loss dataset deep-learning image-classification l1 l2 linear-classifier loss-functions machine-learning matplotlib mean-square-error mnist negative-log-likelihood neural-network numpy python pytorch

Last synced: 7 months ago
JSON representation

Comparison of common loss functions in PyTorch using MNIST dataset

• Host: GitHub
• URL: https://github.com/unvercan/loss-function-comparison-pytorch
• Owner: unvercan
• License: mit
• Created: 2019-10-27T21:13:14.000Z (almost 6 years ago)
• Default Branch: main
• Last Pushed: 2022-02-09T09:45:07.000Z (over 3 years ago)
• Last Synced: 2023-10-07T04:24:33.368Z (about 2 years ago)
• Topics: cross-entropy-loss, dataset, deep-learning, image-classification, l1, l2, linear-classifier, loss-functions, machine-learning, matplotlib, mean-square-error, mnist, negative-log-likelihood, neural-network, numpy, python, pytorch
• Language: Python
• Homepage:
• Size: 3.02 MB
• Stars: 2
• Watchers: 2
• Forks: 0
• Open Issues: 0
• Metadata Files:
  • Readme: README.md
  • License: LICENSE

Awesome Lists containing this project

README

          
# loss-function-comparison-pytorch

Comparison of common loss functions on MNIST dataset using in PyTorch. 

## Loss functions:

- L1

- Mean Squared Error (MSE)

- Negative Log Likelihood (NLL)

- Cross Entropy Loss

## Best result: Cross Entropy Loss