MLProject add

2022-05-15 11:23:25 +02:00 · 2022-05-15 11:23:25 +02:00 · 17379f6457
commit 17379f6457
parent 83a7308d1c
1 changed files with 12 additions and 0 deletions
--- a/12
+++ b/12
@ -0,0 +1,12 @@
+name: s478841 regression model
+
+docker_env:
+  image: s478841-image:latest
+
+entry_points:
+  main:
+    parameters:
+      epochs: { type: string, default: "140" }
+      steps: { type: string, default: "10" }
+      save_model: { type: string, default: "--save" }
+    command: "python3 scripts/mlflow_train.py -e {epochs} -s {steps} {save_model}"