From 6918c0b7e8f18bc51d9816fcfb2144a46be53c05 Mon Sep 17 00:00:00 2001
From: Wojciech Lidwin <59863785+Halal37@users.noreply.github.com>
Date: Fri, 12 May 2023 15:36:50 +0200
Subject: [PATCH] MLproject

---
 MLproject.txt | 16 ++++++++++++++++
 1 file changed, 16 insertions(+)
 create mode 100644 MLproject.txt

diff --git a/MLproject.txt b/MLproject.txt
new file mode 100644
index 0000000..5bf82a3
--- /dev/null
+++ 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}"
\ No newline at end of file