21 lines
454 B
Docker
21 lines
454 B
Docker
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"]
|