diff --git a/dvc.yaml b/dvc.yaml
new file mode 100644
index 0000000..a88b942
--- /dev/null
+++ b/dvc.yaml
@@ -0,0 +1,17 @@
+stages:
+  split_data:
+    cmd: python get_dataset.py
+    deps:
+      - Student_Performance.csv
+    outs:
+      - dataset.csv
+      - df_train.csv
+      - df_test.csv
+
+  train_model:
+    cmd: python train.py 30
+    deps:
+      - df_train.csv
+      - df_test.csv
+    outs:
+      - model.keras