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)