diff --git a/mlflow/Dockerfile b/mlflow/Dockerfile new file mode 100644 index 0000000..08c1332 --- /dev/null +++ b/mlflow/Dockerfile @@ -0,0 +1,15 @@ +FROM python:3.10 + +RUN pip install --upgrade pip + +RUN pip3 install mlflow +RUN pip3 install scikit-learn +RUN pip3 install pandas +RUN pip3 install numpy +RUN pip3 install torch + +COPY mlflow_model.py . +COPY mlflow_prediction.py . +COPY forest_test.csv . +COPY forest_train.csv . +COPY forest_val.csv . diff --git a/mlflow/MLProject b/mlflow/MLProject index 186abe5..d65a38a 100644 --- a/mlflow/MLProject +++ b/mlflow/MLProject @@ -1,8 +1,8 @@ name: mlflow_464914 -conda_env: conda.yaml #ścieżka do pliku conda.yaml z definicją środowisk -# docker_env: -# image: mlflow-docker-example-environment +# conda_env: conda.yaml #ścieżka do pliku conda.yaml z definicją środowisk +docker_env: + image: mlflow_image entry_points: main: