unvercan / activation-function-comparison-pytorch

Comparison of common activation functions on MNIST dataset using PyTorch.

python pytorch image-classification neural-network deep-learning mnist dataset machine-learning relu sigmoid tanh activation-functions matplotlib numpy

activation-functions-comparison-pytorch

Comparison of common activation functions on MNIST dataset using PyTorch.

Activation functions:

Relu
Sigmoid
Tanh

Best result: Relu

About

Comparison of common activation functions on MNIST dataset using PyTorch.

python pytorch image-classification neural-network deep-learning mnist dataset machine-learning relu sigmoid tanh activation-functions matplotlib numpy

MIT License

Languages

Language:Python 100.0%