test
This commit is contained in:
parent
88514fa942
commit
129c498b2d
12
Dockerfile
Normal file
12
Dockerfile
Normal 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 ./
|
@ -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)
|
||||
|
Loading…
Reference in New Issue
Block a user