20 lines
396 B
Docker
20 lines
396 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
|
|
|
|
WORKDIR app
|
|
|
|
ARG EPOCHS
|
|
ENV EPOCHS=${EPOCHS}
|
|
|
|
COPY mlflow_training.py ./
|
|
COPY heart_2020_cleaned.csv ./
|
|
|
|
|
|
CMD ["python3", "./mlflow_training.py $EPOCHS"]
|