This commit is contained in:
Jakub Zaręba 2023-05-10 21:00:09 +02:00
parent ef8c96a780
commit 176f7aa4fe

View File

@ -2,7 +2,7 @@ FROM ubuntu
RUN apt-get update && apt-get install -y python3 python3-pip unzip
RUN python3 -m pip install pandas numpy tensorflow imbalanced-learn sklearn
RUN python3 -m pip install pandas numpy tensorflow imbalanced-learn sklearn sacred
RUN apt-get install -y git
COPY train.py /app/train.py
@ -12,6 +12,7 @@ COPY data.csv /app/data.csv
WORKDIR /app
export SACRED_IGNORE_GIT=TRUE
RUN python3 train.py
RUN python3 predictions.py