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

WORKDIR app

ARG EPOCHS
ENV EPOCHS=${EPOCHS}

COPY ml_training.py ./
COPY heart_2020_cleaned.csv ./


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