28 lines
549 B
Docker
28 lines
549 B
Docker
FROM ubuntu:latest
|
|
|
|
WORKDIR /
|
|
|
|
ARG KAGGLE_USERNAME
|
|
ARG KAGGLE_KEY
|
|
ENV KAGGLE_USERNAME=${KAGGLE_USERNAME}
|
|
ENV KAGGLE_KEY=${KAGGLE_KEY}
|
|
|
|
RUN apt install python3-pip -y
|
|
|
|
RUN pip install --user kaggle
|
|
RUN pip install --user pandas
|
|
RUN pip install --user sklearn
|
|
RUN pip install --user torch
|
|
RUN pip install --user tqdm
|
|
RUN pip install --user seaborn
|
|
|
|
COPY KaggleV2-May-2016.csv ./
|
|
COPY create_data.py ./
|
|
COPY stats_data.py ./
|
|
COPY train_model.py ./
|
|
|
|
CMD ["python", "train_model.py"]
|
|
|
|
# CMD ["python", "create_data.py"]
|
|
# CMD ["python", "stats_data.py"]
|