Skip to content

KohlerHECTOR/MNIST-DeepLearning

Repository files navigation

Deep Learning for image classification

pip install -r  http://webia.lip6.fr/~baskiotisn/requirements-amal.txt

Train an autoencoder

python3 train_autoencoder.py

Train a classical MLP for classification

python3 classif.py

Train a Highway network for classification

python3 train_highway.py

About

Code image classification of MNIST dataset using different architectures: simple linear NN, autoencoder, and highway network

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages