log params with mlflow

.gitignore

@@ -152,4 +152,6 @@ fabric.properties
 
 # kaggle
 kaggle.json
+Car_Prices_Poland_Kaggle*
+CarPrices*
 

Dockerfile

@@ -24,6 +24,7 @@ RUN python3 -m pip install sklearn
 RUN python3 -m pip install matplotlib
 RUN python3 -m pip install sacred
 RUN python3 -m pip install pymongo
+RUN python3 -m pip install mlflow
 RUN python3 -m pip freeze
 
 ENV PATH="/root/.local/bin:${PATH}"

MLproject

@@ -0,0 +1,13 @@
+name: tutorial
+
+conda_env: conda.yaml #ścieżka do pliku conda.yaml z definicją środowiska
+
+#docker_env:
+#  image: mlflow-docker-example-environment
+
+entry_points:
+  main:
+    parameters:
+      epochs: {type: float, default: 0.5}
+      l1_ratio: {type: float, default: 0.1}
+    command: "python train.py {epochs}"