Skip to content

rako233/SimpleMNISTNN

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

9 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Simple Neural Network for MNIST data

I made this Jupyter notebook to learn some concepts of Keras.

  • Showing the progress of the training in a graph
  • Easy configuration of hidden layers
  • Normalizing the input data
  • Initializing the weights with the Xavier function
  • Regularizing the weights with a L2 regularizer
  • Plotting accuracy and loss under training
  • Using tensorboard to monitor the training
  • Using different learning rates while training

The adaptation of the learning rate to the slope of the loss function is paying off. Just compare these examples monitored with tensorboard:

The blue kurves are from the run with a shifting learning rate. It allows to start with a larger learning rate and reducing the learning rate towards the end.

The hyperparameters are optimized for GPU. I installed Anaconda, Cuda9 and Keras on Ubuntu 16.04 to do this experiment.

About

A Very Simple Neural Network for MNIST Data

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published