MLproject

MLproject.txt

@@ -0,0 +1,16 @@
+name: s487197
+
+docker_env:
+  image: s487197/ium:55
+
+entry_points:
+  main:
+    parameters:
+      epochs: {type: int, default: 20}
+      lr: {type: float, default: 0.01}
+      validation_split: {type: float, default: 0.2}
+    command: "python ium_train.py {epochs} {lr} {validation_split}"
+  test:
+    parameters:
+      build: {type: int, default: 1}
+    command: "python ium_predict.py {build}"