Skip to content
/ loss-function-comparison-pytorch Public

Comparison of common loss functions in PyTorch using MNIST dataset

License

MIT license
3 stars 0 forks Branches Tags Activity
Star
Notifications You must be signed in to change notification settings

unvercan/loss-function-comparison-pytorch

BranchesTags

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

14 Commits

array

array

 
 

epoch

epoch

 
 

graph

graph

 
 

.gitignore

.gitignore

 
 

LICENSE

LICENSE

 
 

README.md

README.md

 
 

common.py

common.py

 
 

mnist_classifier_crossentropyloss.py

mnist_classifier_crossentropyloss.py

 
 

mnist_classifier_l1loss.py

mnist_classifier_l1loss.py

 
 

mnist_classifier_mseloss.py

mnist_classifier_mseloss.py

 
 

mnist_classifier_nnloss.py

mnist_classifier_nnloss.py

 
 

Repository files navigation

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

About

Comparison of common loss functions in PyTorch using MNIST dataset

Topics

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

Resources

Readme

License

MIT license
Activity

Stars

3 stars

Watchers

2 watching

Forks

0 forks
Report repository

Languages