This commit is contained in:
Jakub Zaręba 2023-05-10 23:42:46 +02:00
parent 6c00b125aa
commit 2dffac65ab
1 changed files with 3 additions and 22 deletions

View File

@ -1,22 +1,6 @@
FROM ubuntu:20.04
FROM ubuntu
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 apt-get update && apt-get install -y python3 python3-pip unzip
RUN python3 -m pip install pandas numpy tensorflow imbalanced-learn sklearn sacred pymongo mlflow
RUN apt-get install -y git
@ -28,9 +12,6 @@ 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
RUN python3 train.py --epochs 10
CMD ["python3", "predictions.py"]