Add MLproject for MLFlow

.gitignore

@@ -20,3 +20,4 @@ evaluation_results.txt
 model_out
 trend.png
 sacred_runs
+mlruns

MLproject

@@ -0,0 +1,14 @@
+name: ium_s444409
+
+docker_env:
+  image: s444409-create-dataset
+  volumes: ["/tmp/mlruns:/mlruns:rw"]
+
+entry_points:
+  main:
+    parameters:
+      epochs: {type: float, default: 5}
+      batch_size: {type: float, default: 64}
+    command: "python train_model.py -e {epochs} -b {batch_size}"
+  eval:
+    command: "python eval_model.py"

train_model.py

@@ -12,7 +12,7 @@ default_epochs = 5
 
 device = "cuda" if torch.cuda.is_available() else "cpu"
 
-mlflow.set_tracking_uri("http://172.17.0.1:5000")
+#mlflow.set_tracking_uri("http://172.17.0.1:5000")
 mlflow.set_experiment("s444409")