20 lines
380 B
Docker
20 lines
380 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 CUTOFF
|
|
ENV CUTOFF=${CUTOFF}
|
|
|
|
COPY ml_training.py ./
|
|
COPY heart_2020_cleaned.csv ./
|
|
|
|
|
|
CMD ["python3", "./ml_training.py"]
|