From 1b646836f870b7ac7023fbd1324696fe2473b27c Mon Sep 17 00:00:00 2001
From: Sheaza <nataliawolowiec3@gmail.com>
Date: Tue, 14 May 2024 23:48:41 +0200
Subject: [PATCH] add conda env

---
 MLproject  |  9 +++++++++
 conda.yaml | 13 +++++++++++++
 2 files changed, 22 insertions(+)
 create mode 100644 MLproject
 create mode 100644 conda.yaml

diff --git a/MLproject b/MLproject
new file mode 100644
index 0000000..64ccc44
--- /dev/null
+++ b/MLproject
@@ -0,0 +1,9 @@
+name: s464980
+
+conda_env: conda.yaml
+
+entry_points:
+  optimal_parameters:
+    parameters:
+      epochs: { type: int, default: 20 }
+    command: 'python train.py {epochs}'
\ No newline at end of file
diff --git a/conda.yaml b/conda.yaml
new file mode 100644
index 0000000..f4dfd5b
--- /dev/null
+++ b/conda.yaml
@@ -0,0 +1,13 @@
+name: s464980
+channels:
+  - defaults
+dependencies:
+  - python=3.11
+  - pip
+  - pip:
+      - mlflow
+      - tensorflow
+      - pandas
+      - scikit-learn
+      - numpy
+      - matplotlib
\ No newline at end of file