Update 'dvc.yaml'

2022-05-30 21:09:04 +02:00 · 2022-05-30 21:09:04 +02:00 · 5cab54fd20
commit 5cab54fd20
parent 6a8f4aa868
1 changed files with 11 additions and 1 deletions
--- a/dvc.yaml
+++ b/dvc.yaml
@ -3,6 +3,16 @@ stages:
    cmd: python3 process_dataset.py
    deps:
    - lego_sets.csv
+    outs:
+    - lego_sets_clean.csv
+    - lego_sets_clean_train.csv
+    - lego_sets_clean_valid.csv
+    - lego_sets_clean_test.csv
  train:
    cmd: python3 simple_regression.py 100
-  
+    deps:
+    - lego_sets_clean_train.csv
+    - lego_sets_clean_test.csv
+    outs:
+    - lego_reg_model
+    - lego_reg_results.csv