37 lines
933 B
Docker
37 lines
933 B
Docker
FROM ubuntu:20.04
|
|
|
|
RUN apt-get update && apt-get install -y curl git libssl-dev make build-essential libbz2-dev \
|
|
libreadline-dev libsqlite3-dev wget curl llvm libncurses5-dev libncursesw5-dev \
|
|
xz-utils tk-dev libffi-dev liblzma-dev python3-openssl git
|
|
|
|
RUN curl https://pyenv.run | bash
|
|
|
|
ENV HOME /root
|
|
ENV PYENV_ROOT $HOME/.pyenv
|
|
ENV PATH $PYENV_ROOT/shims:$PYENV_ROOT/bin:$PATH
|
|
ENV PYTHON_VERSION 3.11.3
|
|
|
|
RUN pyenv install $PYTHON_VERSION
|
|
RUN pyenv global $PYTHON_VERSION
|
|
|
|
RUN python --version
|
|
|
|
RUN apt-get install -y python3-pip unzip
|
|
|
|
RUN python3 -m pip install pandas numpy tensorflow imbalanced-learn sklearn sacred pymongo mlflow
|
|
RUN apt-get install -y git
|
|
|
|
COPY train.py /app/train.py
|
|
COPY predictions.py /app/predictions.py
|
|
COPY data.csv /app/data.csv
|
|
|
|
WORKDIR /app
|
|
|
|
RUN export SACRED_IGNORE_GIT=TRUE
|
|
|
|
SHELL ["/bin/bash", "-c"]
|
|
|
|
RUN source ~/.bashrc && mlflow run . -P epochs=10
|
|
|
|
CMD ["python3", "predictions.py"]
|