18 lines
421 B
Docker
18 lines
421 B
Docker
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 sacred
|
|
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
|
|
|
|
export SACRED_IGNORE_GIT=TRUE
|
|
RUN python3 train.py
|
|
|
|
RUN python3 predictions.py |