Skip to content

Commit f7c75a4

Browse files
committed
Add support for Docker
1 parent 2ae2a24 commit f7c75a4

2 files changed

Lines changed: 29 additions & 1 deletion

File tree

Dockerfile

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,9 @@
1+
FROM python
2+
3+
RUN pip install numpy matplotlib
4+
5+
WORKDIR /usr/src/app
6+
7+
COPY . .
8+
9+
CMD ["python", "src/digits.py"]

README.md

Lines changed: 20 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,10 +2,29 @@
22
[![codecov](https://codecov.io/gh/veetimar/neural-network-python/graph/badge.svg?token=B3NIJQEDA6)](https://codecov.io/gh/veetimar/neural-network-python)
33

44
## About
5+
56
A fully connected neural network, ELU activation in hidden layers and sigmoid in output layer.
67
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+
79
## 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.
18+
19+
### The hard way (Poetry and Python installed)
20+
21+
Clone the repository
22+
23+
```console
24+
git clone https://github.com/veetimar/neural-network-python
25+
```
26+
27+
Install the required dependencies with
928

1029
```console
1130
poetry install --without dev

0 commit comments

Comments
 (0)