mlflow

MLproject.yml

@@ -0,0 +1,16 @@
+name: s487194
+
+docker_env:
+  image: s487194/ium:1
+
+entry_points:
+  main:
+    parameters:
+      epochs: {type: int, default: 100}
+      lr: {type: float, default: 0.01}
+    command: "python train.py --epochs {epochs} --lr {lr}"
+  test:
+    parameters:
+      
+    command: "python prediction.py"
+