diff --git a/dvc.yaml b/dvc.yaml
new file mode 100644
index 0000000..891946e
--- /dev/null
+++ b/dvc.yaml
@@ -0,0 +1,11 @@
+stages:
+  train:
+    cmd: python ./train.py 10
+    deps:
+    - Customers.csv
+    - train.py
+  test:
+    cmd: python ./test.py
+    deps:
+    - cifar_net.pth
+    - test.py