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 apt-get install unzip

WORKDIR /app


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