Create dvc.yaml

2024-05-28 21:43:31 +02:00 · 2024-05-28 21:43:31 +02:00 · 0f254aa5fa
commit 0f254aa5fa
parent abb213675e
1 changed files with 29 additions and 0 deletions
--- a/dvc.yaml
+++ 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