all done with plots

shivpalSW · Jul 16, 2023 · be59d89 · be59d89
1 parent 7549722
commit be59d89
Show file tree

Hide file tree

Showing 4 changed files with 10,574 additions and 14 deletions.
diff --git a/dvc.lock b/dvc.lock
@@ -4,17 +4,17 @@ stages:
     cmd: python src/stage_01_prepare.py --config=configs/config.yaml --params=params.yaml
     deps:
     - path: configs/config.yaml
-      md5: 508d0ae144ca6fa812180f6d0057a3f2
-      size: 374
+      md5: 28c91b4608db6b4d5754223f44a81a31
+      size: 453
     - path: data/data.xml
       md5: a304afb96060aad90176268345e10355
       size: 37891850
     - path: src/stage_01_prepare.py
       md5: 47a1121c82f1e79ff3004abb9c4d5d4d
       size: 2096
     - path: src/utils/common.py
-      md5: ebe79d372c0b5256c6dacc26b27a7acf
-      size: 840
+      md5: a9bd2172b6d5d14931e59c0fad897a58
+      size: 979
     - path: src/utils/data_mgmt.py
       md5: c985e579010a81c5462ebf8184649fe4
       size: 871
@@ -39,14 +39,14 @@ stages:
       md5: 553034c4cf40efc63c99c19fe98610d0
       size: 18986541
     - path: configs/config.yaml
-      md5: 508d0ae144ca6fa812180f6d0057a3f2
-      size: 374
+      md5: 28c91b4608db6b4d5754223f44a81a31
+      size: 453
     - path: src/stage_02_featurization.py
       md5: e07203a587149c1f889134b58d8112a3
       size: 2991
     - path: src/utils/common.py
-      md5: ebe79d372c0b5256c6dacc26b27a7acf
-      size: 840
+      md5: a9bd2172b6d5d14931e59c0fad897a58
+      size: 979
     - path: src/utils/featurize.py
       md5: 91d515f8a5c06f0d0c055c0f3b6b3125
       size: 516
@@ -68,14 +68,14 @@ stages:
       md5: ecc944089c3a29ca33e6cfd158f81b0d
       size: 10134795
     - path: configs/config.yaml
-      md5: 508d0ae144ca6fa812180f6d0057a3f2
-      size: 374
+      md5: 28c91b4608db6b4d5754223f44a81a31
+      size: 453
     - path: src/stage_03_train.py
       md5: bb5dcfadc16299ec07fd19fd9ceba2fd
       size: 2223
     - path: src/utils/common.py
-      md5: ebe79d372c0b5256c6dacc26b27a7acf
-      size: 840
+      md5: a9bd2172b6d5d14931e59c0fad897a58
+      size: 979
     params:
       params.yaml:
         train.min_split: 16
@@ -85,3 +85,33 @@ stages:
     - path: artifacts/model/model.pkl
       md5: 20ac77b4d7237b3cf0c6e676bdc7dbbc
       size: 3558016
+  evaluate:
+    cmd: python src/stage_04_evaluate.py --config=configs/config.yaml
+    deps:
+    - path: artifacts/features/test.pkl
+      md5: a16444f35be763056c35610d6741fa00
+      size: 2597499
+    - path: artifacts/model/model.pkl
+      md5: 20ac77b4d7237b3cf0c6e676bdc7dbbc
+      size: 3558016
+    - path: configs/config.yaml
+      md5: 28c91b4608db6b4d5754223f44a81a31
+      size: 453
+    - path: src/stage_04_evaluate.py
+      md5: df4e36d182f823c42b4a3b42b8c44c25
+      size: 2832
+    - path: src/utils/common.py
+      md5: a9bd2172b6d5d14931e59c0fad897a58
+      size: 979
+    outs:
+    - path: prc.json
+      md5: 28dedc46a5008794a2514dc4f1e4b3ea
+      size: 130624
+    - path: roc.json
+      md5: 577400ce2214fe678f08d344059a6d7c
+      size: 95592
+    - path: scores.json
+      md5: 075b99c7b6e7123f84c0292a80876f0f
+      size: 75
+  run_plots_command:
+    cmd: dvc plots diff
diff --git a/dvc.yaml b/dvc.yaml
@@ -43,4 +43,28 @@ stages:
       - train.n_est
       - train.min_split
     outs:
-      - artifacts/model/model.pkl      
+      - artifacts/model/model.pkl
+
+  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
+
+  run_plots_command:
+    cmd: dvc plots diff