FROM ubuntu

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

RUN python3 -m pip install pandas numpy tensorflow imbalanced-learn sklearn
RUN apt-get install -y git

COPY train.py /app/train.py
COPY predictions.py /app/predictions.py
COPY data.csv /app/data.csv
# COPY test_data.csv /app/test_data.csv

WORKDIR /app

RUN python3 train.py

RUN python3 predictions.py