ium_487187/Dockerfile

18 lines
420 B
Docker
Raw Normal View History

2023-05-10 23:42:46 +02:00
FROM ubuntu
2023-04-04 19:46:06 +02:00
2023-05-10 23:42:46 +02:00
RUN apt-get update && apt-get install -y python3 python3-pip unzip
2023-05-10 23:39:09 +02:00
RUN python3 -m pip install pandas numpy tensorflow imbalanced-learn sklearn sacred pymongo mlflow
RUN apt-get install -y git
2023-05-07 21:10:11 +02:00
2023-05-10 14:11:16 +02:00
COPY train.py /app/train.py
COPY predictions.py /app/predictions.py
COPY data.csv /app/data.csv
2023-05-07 21:10:11 +02:00
2023-05-10 14:11:16 +02:00
WORKDIR /app
2023-04-04 19:46:06 +02:00
2023-05-10 21:02:34 +02:00
RUN export SACRED_IGNORE_GIT=TRUE
2023-05-10 23:42:46 +02:00
RUN python3 train.py --epochs 10
2023-05-10 23:33:25 +02:00
CMD ["python3", "predictions.py"]