From 129c498b2d6bf89d37e62f3867860724c273311e Mon Sep 17 00:00:00 2001 From: szymonj98 Date: Wed, 27 Apr 2022 19:51:21 +0200 Subject: [PATCH] test --- Dockerfile | 12 ++++++++++++ biblioteki_dl.py | 2 +- 2 files changed, 13 insertions(+), 1 deletion(-) create mode 100644 Dockerfile diff --git a/Dockerfile b/Dockerfile new file mode 100644 index 0000000..7f87558 --- /dev/null +++ b/Dockerfile @@ -0,0 +1,12 @@ +FROM ubuntu:latest +WORKDIR /ium +RUN apt update && apt install -y python3-pip +RUN apt install unzip +RUN pip3 install kaggle +RUN mkdir /.kaggle && chmod o+w /.kaggle +RUN pip3 install pandas +RUN pip3 install numpy +RUN pip3 install sklearn +RUN pip3 install tensorflow +COPY ./steam-200k.csv ./ +COPY ./biblioteki_dl.py ./ \ No newline at end of file diff --git a/biblioteki_dl.py b/biblioteki_dl.py index 4716b5e..799f7ad 100644 --- a/biblioteki_dl.py +++ b/biblioteki_dl.py @@ -102,7 +102,7 @@ y_test = np.array(y_test).astype(np.float32) -model.fit(x_train, y_train, epochs=100) +model.fit(x_train, y_train, epochs=2) model.evaluate(x_test, y_test) prediction = model.predict(x_test) classes_x=np.argmax(prediction,axis=1)