updated dvc.yaml file

shivpalSW · Jul 17, 2023 · 3820bf2 · 3820bf2
1 parent b87a775
commit 3820bf2
Showing 1 changed file with 36 additions and 31 deletions.
diff --git a/dvc.yaml b/dvc.yaml
@@ -14,6 +14,7 @@ stages:
       outs:
         - artifacts/prepared/train.tsv
         - artifacts/prepared/test.tsv
+      reflink: false
 
   featurize:
       cmd: python src/stage_02_featurization.py --config=configs/config.yaml --params=params.yaml
@@ -30,41 +31,45 @@ stages:
       outs:
         - artifacts/features/train.pkl
         - artifacts/features/test.pkl
+      reflink: false
 
   train:
-    cmd: python src/stage_03_train.py --config=configs/config.yaml --params=params.yaml
-    deps:
-      - src/stage_03_train.py
-      - artifacts/features/train.pkl
-      - src/utils/common.py
-      - configs/config.yaml
-    params:
-      - train.seed
-      - train.n_est
-      - train.min_split
-    outs:
-      - artifacts/model/model.pkl
+      cmd: python src/stage_03_train.py --config=configs/config.yaml --params=params.yaml
+      deps:
+        - src/stage_03_train.py
+        - artifacts/features/train.pkl
+        - src/utils/common.py
+        - configs/config.yaml
+      params:
+        - train.seed
+        - train.n_est
+        - train.min_split
+      outs:
+        - artifacts/model/model.pkl
+      reflink: false
 
   evaluate:
-    cmd: python src/stage_04_evaluate.py --config=configs/config.yaml
-    deps:
-      - src/stage_04_evaluate.py
-      - artifacts/features/test.pkl
-      - src/utils/common.py
-      - configs/config.yaml
-      - artifacts/model/model.pkl
-    metrics:
-      - scores.json:
-          cache: false
-    plots:
-      - prc.json:
-          cache: false
-          x: recall
-          y: precision
-      - roc.json:
-          cache: false
-          x: fpr
-          y: tpr
+      cmd: python src/stage_04_evaluate.py --config=configs/config.yaml
+      deps:
+        - src/stage_04_evaluate.py
+        - artifacts/features/test.pkl
+        - src/utils/common.py
+        - configs/config.yaml
+        - artifacts/model/model.pkl
+      metrics:
+        - scores.json:
+            cache: false
+      plots:
+        - prc.json:
+            cache: false
+            x: recall
+            y: precision
+        - roc.json:
+            cache: false
+            x: fpr
+            y: tpr
+
+      reflink: false
 
   run_plots_command:
     cmd: dvc plots diff