Fix MLProject

MLProject

@@ -1,15 +1,14 @@
 name: HousePriceModel
 
-docker_env:
+conda_env: environment.yml
-  image: paweldopierala/ium:2.0.0
 
 entry_points:
   main:
     parameters:
       epochs: {type: int, default: 20}
       learning_rate: {type: float, default: 0.001}
-      batch_size: {type: int, default: 20}
+      batch_size: {type: int, default: 32}
-    command: "python train.py {epochs} {learning_rate} {batch_size}"
+    command: "python create_model.py {epochs} {learning_rate} {batch_size}"
 
   test:
-    command: "python test.py"
+    command: "python evaluate.py"

environment.yml

@@ -0,0 +1,13 @@
+name: hp_model_env
+channels:
+  - defaults
+dependencies:
+  - python
+  - pandas
+  - numpy
+  - scikit-learn
+  - matplotlib
+  - tensorflow
+  - keras
+  - mlflow
+  - pip