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

COPY ml_prepare.py ./
COPY ml_training.py ./
RUN useradd -r -u 111 s444465

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