diff --git a/dvc.yaml b/dvc.yaml
new file mode 100644
index 0000000..86e9681
--- /dev/null
+++ b/dvc.yaml
@@ -0,0 +1,7 @@
+stages:
+    data:
+        cmd: ./data/data.sh
+    train@0:
+        cmd: python ./learning/ml-mlflow.py
+    train@1:
+        cmd: python ./learning/ml.py