You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
A fully connected neural network, ELU activation in hidden layers and sigmoid in output layer.
6
7
Comes with a simple user interface for training and testing the neural net with MNIST hand-written digits but can be used for practically any task since the size of the neural net and batch size can be freely choosed. Is able to achieve over 97% accuracy in the MNIST dataset with just a few minutes of training.
8
+
7
9
## How to use
8
-
Clone the repository and install the required dependencies with:
10
+
11
+
### The easy way (Docker installed)
12
+
13
+
```console
14
+
docker run -it veetimar/neural-network
15
+
```
16
+
17
+
Note that matplotlib seems not to work inside a container so plotting error after training and showing images classified wrong/right do not work when using Docker.
0 commit comments