ium_444380/Dockerfile
2022-05-03 19:47:36 +02:00

21 lines
489 B
Docker

FROM ubuntu
ARG KAGGLE_USERNAME
ARG KAGGLE_KEY
ARG CUTOFF
ENV CUTOFF=${CUTOFF:-260000}
ENV IS_DOCKER=True
WORKDIR /app
COPY ./download_data.sh calc_stats.sh ./
COPY ./clean_and_split_data.py calc_stats.py ./
COPY ./train_model.py ./
RUN apt-get update && apt-get install -y python3-pip unzip && rm -rf /var/lib/apt/lists/*
RUN export PATH="$PATH:/root/.local/bin"
RUN pip3 install kaggle pandas scikit-learn tensorflow keras matplotlib numpy
RUN mkdir /.kaggle && chmod o+w /.kaggle