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 install -y python3 python3-pip
RUN python3 -m pip install --user kaggle
RUN python3 -m pip install --user pandas
RUN python3 -m pip install --user scikit-learn
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
WORKDIR app
ARG CUTOFF
ENV CUTOFF=${CUTOFF}
COPY main_docker.py ./
COPY ml_training.py ./
COPY heart_2020_cleaned.csv ./
CMD ["python3", "./main_docker.py"]
CMD ["python3", "./ml_training.py"]