Dockerfile modified

This commit is contained in:
Andrzej Preibisz 2022-04-07 22:12:11 +02:00
parent 22614d0a3e
commit 97b7cbca27

View File

@ -2,17 +2,18 @@ FROM ubuntu:latest
RUN apt-get update -y RUN apt-get update -y
RUN apt-get install -y python3 python3-pip RUN apt-get install -y python3 python3-pip
RUN python3 -m pip install --user kaggle RUN python3 -m pip install kaggle
RUN python3 -m pip install --user pandas RUN python3 -m pip install pandas
RUN python3 -m pip install --user scikit-learn RUN python3 -m pip install scikit-learn
RUN python3 -m pip install --upgrade tensorflow
WORKDIR app WORKDIR app
ARG CUTOFF ARG CUTOFF
ENV CUTOFF=${CUTOFF} ENV CUTOFF=${CUTOFF}
COPY main_docker.py ./ COPY ml_training.py ./
COPY heart_2020_cleaned.csv ./ COPY heart_2020_cleaned.csv ./
CMD ["python3", "./main_docker.py"] CMD ["python3", "./ml_training.py"]