From c4ce89938cf68e2508c5aff46586ac4ca11deb88 Mon Sep 17 00:00:00 2001
From: Alicja Szulecka <73056579+AliSzu@users.noreply.github.com>
Date: Mon, 6 May 2024 22:08:05 +0200
Subject: [PATCH] mlflow

---
 mlflow/Dockerfile | 15 +++++++++++++++
 mlflow/MLProject  |  6 +++---
 2 files changed, 18 insertions(+), 3 deletions(-)
 create mode 100644 mlflow/Dockerfile

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: