add conda env

2024-05-14 23:48:41 +02:00 · 2024-05-14 23:48:41 +02:00 · 1b646836f8
commit 1b646836f8
parent 8ba3d2b97c
2 changed files with 22 additions and 0 deletions
--- a/9
+++ b/9
@ -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}'
--- a/conda.yaml
+++ b/conda.yaml
@ -0,0 +1,13 @@
+name: s464980
+channels:
+  - defaults
+dependencies:
+  - python=3.11
+  - pip
+  - pip:
+      - mlflow
+      - tensorflow
+      - pandas
+      - scikit-learn
+      - numpy
+      - matplotlib