FROM ubuntu:latest

WORKDIR /

ARG KAGGLE_USERNAME
ARG KAGGLE_KEY
ENV KAGGLE_USERNAME=${KAGGLE_USERNAME}
ENV KAGGLE_KEY=${KAGGLE_KEY}

RUN apt update
RUN apt-get update
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"]