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 ./