From 0f254aa5fa8af58caa837631fe62d7c7e13bd29f Mon Sep 17 00:00:00 2001
From: Alicja Szulecka <73056579+AliSzu@users.noreply.github.com>
Date: Tue, 28 May 2024 21:43:31 +0200
Subject: [PATCH] Create dvc.yaml

---
 dvc.yaml | 29 +++++++++++++++++++++++++++++
 1 file changed, 29 insertions(+)
 create mode 100644 dvc.yaml

diff --git a/dvc.yaml b/dvc.yaml
new file mode 100644
index 0000000..c3f2511
--- /dev/null
+++ b/dvc.yaml
@@ -0,0 +1,29 @@
+stages:
+  prepare_data:
+    cmd: python ./IUM_2.py
+    deps:
+      - create-dataset.py
+      - covtype.csv
+    outs:
+      - forest_train.csv
+      - forest_test.csv
+      - forest_val.csv
+
+  train_model:
+    cmd: python ./model.py
+    deps:
+      - model.py
+      - forest_train.csv
+      - forest_test.csv
+      - forest_val.csv
+    outs:
+      - model.pth
+
+  evaluate_model:
+    cmd: python ./prediction.py
+    deps:
+      - prediction.py
+      - model.pth
+      - forest_test.csv
+    outs:
+      - predictions.txt
\ No newline at end of file