diff --git a/dvc.yaml b/dvc.yaml
new file mode 100644
index 0000000..649c7df
--- /dev/null
+++ b/dvc.yaml
@@ -0,0 +1,8 @@
+stages:
+  prepare:
+    cmd: python3 process_dataset.py
+    deps:
+    - lego_sets.csv
+  train:
+    cmd: python3 simple_regression.py 100
+