Add dvc.yaml

2022-06-05 21:58:57 +02:00 · 2022-06-05 21:58:57 +02:00 · 73b627f6ee
commit 73b627f6ee
parent 7125eeac61
3 changed files with 21 additions and 1 deletions
--- a/dvc.yaml
+++ b/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
--- a/model_params.yaml
+++ b/model_params.yaml
@ -0,0 +1,3 @@
+train:
+  batch_size: 64
+  epochs: 5
--- a/requirements.txt
+++ b/requirements.txt
@ -6,3 +6,6 @@ matplotlib==3.5.2
 sacred==0.8.2
 pymongo==4.1.1
 mlflow==1.25.1
+dvc==2.10.2
+dvc-ssh==0.0.1a0
+paramiko==2.11.0