From 17379f645714ea61d30f439abd6f1112f356acff Mon Sep 17 00:00:00 2001
From: MatOgr <ogrodowczykmat@gmail.com>
Date: Sun, 15 May 2022 11:23:25 +0200
Subject: [PATCH] MLProject add

---
 MLProject | 12 ++++++++++++
 1 file changed, 12 insertions(+)
 create mode 100644 MLProject

diff --git a/MLProject b/MLProject
new file mode 100644
index 0000000..48a4f36
--- /dev/null
+++ b/MLProject
@@ -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}"