Add dvc.yaml

dvc.yaml

@@ -0,0 +1,14 @@
+stages:
+  prepare:
+    cmd: ./download_dataset.sh
+    
+  train:
+    cmd: python3 train_model.py
+    deps:
+      - data/Plant_1_Generation_data.csv.train
+    outs:
+      - model_out
+    plots:
+      - trend.png
+    params:
+      - model_params.yaml

model_params.yaml

@@ -0,0 +1,3 @@
+train:
+  batch_size: 64
+  epochs: 5

requirements.txt

@@ -5,4 +5,7 @@ numpy~=1.22.3
 matplotlib==3.5.2
 sacred==0.8.2
 pymongo==4.1.1
-mlflow==1.25.1
+mlflow==1.25.1
+dvc==2.10.2
+dvc-ssh==0.0.1a0
+paramiko==2.11.0