s444380/ium_444380
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Fixed docker
Some checks failed
s444380-training/pipeline/head There was a failure building this commit
Details

Browse Source
This commit is contained in:
Kamil Guttmann 2022-05-03 15:33:07 +02:00
parent e934c1bf17
commit 197008f99d
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
Dockerfile
View File

@ -15,6 +15,6 @@ RUN apt-get update && apt-get install -y python3-pip unzip && rm -rf /var/lib/ap
RUN export PATH="$PATH:/root/.local/bin"
RUN pip3 install kaggle pandas scikit-learn tensorflow
RUN pip3 install kaggle pandas scikit-learn tensorflow keras
RUN mkdir /.kaggle && chmod o+w /.kaggle

2
train_model.py
View File

@ -71,7 +71,7 @@ else:
epochs = 10
# Train model
model.fit(x_train, y_train, epochs=epochs, validation_data=(x_val, y_val))
model.fit(x_train, y_train, epochs=int(epochs), validation_data=(x_val, y_val))
# Make predictions
y_pred = model.predict(x_test)