MLproject

2023-05-12 15:36:50 +02:00 · 2023-05-12 15:36:50 +02:00 · 6918c0b7e8
commit 6918c0b7e8
parent 96e4c03d11
1 changed files with 16 additions and 0 deletions
--- a/MLproject.txt
+++ b/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}"