diff --git a/.gitignore b/.gitignore index bee8a64..8025bc0 100644 --- a/.gitignore +++ b/.gitignore @@ -1 +1,2 @@ __pycache__ +*~ diff --git a/Dockerfile b/Dockerfile new file mode 100644 index 0000000..905c846 --- /dev/null +++ b/Dockerfile @@ -0,0 +1,11 @@ +FROM ubuntu:20.04 + +RUN apt update +RUN apt install -y python3 python3-pip +RUN pip3 install numpy theano + +COPY . /usr/src/dockertest + +WORKDIR /usr/src/dockertest + +CMD ["python3", "test_network1.py"] diff --git a/test_network1.py b/test_network1.py new file mode 100644 index 0000000..1e79346 --- /dev/null +++ b/test_network1.py @@ -0,0 +1,9 @@ +import mnist_loader +import network + +print("gathering data...") +training_data, validation_data, test_data = mnist_loader.load_data_wrapper() +print("data gathered.") +net = network.Network([784, 30, 10]) +print("starting to train network") +net.SGD(training_data, 30, 10, 3.0, test_data=test_data)