Skip to content

Implementation of a simple neural network for image classification purposes.

Notifications You must be signed in to change notification settings

bfreskura/simple_neural_net

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

21 Commits
 
 
 
 
 
 

Repository files navigation

Simple neural network

Implementation of a simple feedforward neural net with one hidden layer.

Dataset

I've used the MNIST dataset.

Results

Results with batch size = 1, no SGD

Epochs Hidden Layer Dimension Precision on the Test set [%] Learning Rate
8 200 94.3 0.001

Results with SGD

Epochs Hidden Layer Dimension Precision on the Test set [%] Learning Rate Minibatch size
30 100 95.3 0.002 200
50 100 95.8 0.002 200
100 100 96.4 0.0009 200
100 200 96.5 0.0009 300

How to use

Run src/simple_net.py

About

Implementation of a simple neural network for image classification purposes.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages