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

WORKDIR /app

COPY ./training.py ./
COPY ./evaluation.py ./