diff --git a/dvc.yaml b/dvc.yaml
new file mode 100644
index 0000000..f92dd9a
--- /dev/null
+++ 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
\ No newline at end of file
diff --git a/model_params.yaml b/model_params.yaml
new file mode 100644
index 0000000..cbf4e69
--- /dev/null
+++ b/model_params.yaml
@@ -0,0 +1,3 @@
+train:
+  batch_size: 64
+  epochs: 5
\ No newline at end of file
diff --git a/requirements.txt b/requirements.txt
index 2aa39c9..417886d 100644
--- a/requirements.txt
+++ b/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
\ No newline at end of file
+mlflow==1.25.1
+dvc==2.10.2
+dvc-ssh==0.0.1a0
+paramiko==2.11.0
\ No newline at end of file