diff --git a/dvc.yaml b/dvc.yaml
new file mode 100644
index 0000000..58a42ae
--- /dev/null
+++ b/dvc.yaml
@@ -0,0 +1,5 @@
+stages:
+  train:
+    cmd: python train.py --epochs=10
+  predict:
+    cmd: python predict.py