README.md

SNN - Simple Neural Network

A Neural Network library written in Rust.

Test 1 Training on MNIST data set with following configuration:

1. Two hidden layers with 50 activation functions (RELU)
2. Softmax classifier
3. Minibatch size 200
4. Learning rate 0.05
5. No regularization
6. Minibatch optimizer

Test 2 Training on MNIST data set with following configuration:

1. Two hidden layers with 50 activation functions (RELU)
2. Softmax classifier
3. Minibatch size 200
4. Learning rate 0.01
5. No regularization
6. Adam optimizer

Test results

• With Adam optimizer we see ~95% accuracy on ~385 iterations
• Without Adam we see ~ 93% accuracy after 1200 iteration.

Thus with Adam optimizer we observe a reach to a slightly higher accuracy ~3.1 times faster.

The library supports:

1. L2 regularization
2. Minibatch, Momentum, RMSProp and Adam optimizations
3. Only Dense layers so far.
4. No GPU acceleration yet