22 lines
549 B
Docker
22 lines
549 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 ./train_model_mlflow.py ./
|
|
COPY ./evaluate.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 sacred mlflow
|
|
|
|
RUN mkdir /.kaggle && chmod o+w /.kaggle
|