This commit is contained in:
szymonj98 2022-04-27 19:51:21 +02:00
parent 88514fa942
commit 129c498b2d
2 changed files with 13 additions and 1 deletions

12
Dockerfile Normal file
View File

@ -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 ./

View File

@ -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)