mlflow

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 .

mlflow/MLProject

@@ -1,8 +1,8 @@
 name: mlflow_464914
 
-conda_env: conda.yaml #ścieżka do pliku conda.yaml z definicją środowisk
+# conda_env: conda.yaml #ścieżka do pliku conda.yaml z definicją środowisk
-# docker_env:
+docker_env:
-#  image: mlflow-docker-example-environment
+ image: mlflow_image
 
 entry_points:
   main: