ium_444465/Dockerfile

FROM ubuntu:latest
RUN apt-get update -y
RUN apt-get install -y python3 python3-pip

RUN python3 -m pip install kaggle
RUN python3 -m pip install pandas
RUN python3 -m pip install scikit-learn
RUN python3 -m pip install --upgrade tensorflow
RUN python3 -m pip install dvc
RUN python3 -m pip install dvc[ssh] paramiko

WORKDIR app

ARG EPOCHS
ENV EPOCHS=${EPOCHS}

COPY ml_prepare.py ./
COPY ml_training.py ./

CMD ["python3", "./ml_training.py $EPOCHS"]