FROM ubuntu:latest

RUN apt update && apt install -y python3-pip

RUN pip3 install kaggle
RUN pip3 install pandas
RUN pip3 install sklearn
RUN pip3 install torch
RUN pip3 install tqdm
RUN pip3 install seaborn
RUN pip3 install dvc
RUN pip3 install dvc[ssh] paramiko

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"]