FROM ubuntu:latest

RUN apt update && apt install -y python3-pip
RUN pip3 install pandas
RUN pip3 install sklearn
RUN pip3 install seaborn
RUN pip3 install ipython
RUN pip3 install torch
RUN pip3 install numpy
RUN pip3 install dvc
RUN apt-get install unzip

WORKDIR /app


COPY ./body-performance-data.zip ./
COPY ./prepare_datasets.py ./
COPY ./train.py ./