upd

.github/workflows/train-model.yaml

@@ -36,7 +36,7 @@ jobs:
         run: |
           pip install -r requirements.txt
           dvc pull
-          dvc repro
+          dvc exp run
           dvc push
           # Create CML report
           echo "## Metrics" >> report.md

dvc.lock

@@ -77,12 +77,12 @@ stages:
           model_path: models/clf-model.joblib
           params:
             n_estimators: 200
-            max_depth: 20
+            max_depth: 21
     outs:
     - path: models/clf-model.joblib
       hash: md5
-      md5: 642d147fb57ad3c13e7d348a4e58a5c1
-      size: 21919560
+      md5: 972585a1016c85ead93ccc2e39d42545
+      size: 22226856
   eval:
     cmd: python src/stages/eval.py --config=params.yaml
     deps:
@@ -93,8 +93,8 @@ stages:
       nfiles: 4
     - path: models/clf-model.joblib
       hash: md5
-      md5: 642d147fb57ad3c13e7d348a4e58a5c1
-      size: 21919560
+      md5: 972585a1016c85ead93ccc2e39d42545
+      size: 22226856
     - path: src/stages/eval.py
       hash: md5
       md5: e402ec2f10cedbc6c02331cf13cd1147
@@ -127,17 +127,17 @@ stages:
           model_path: models/clf-model.joblib
           params:
             n_estimators: 200
-            max_depth: 20
+            max_depth: 21
     outs:
     - path: metrics.json
       hash: md5
-      md5: d09999567bf1267d286f451e89ae31f9
-      size: 67
+      md5: 5c3d801e7c183a768aa1465050b0df13
+      size: 66
     - path: reports/feat_imp.csv
       hash: md5
-      md5: e3f542978e5425e62994b7536cc7cba7
-      size: 186
+      md5: 83deedeb91458076b1408e224c654dda
+      size: 185
     - path: reports/figures/cm.png
       hash: md5
-      md5: 0b64e4a73106f1edb4dea5ebd6a8d3f1
-      size: 10380
+      md5: 454e3f67b0a3b92a581550c1c80d0dd4
+      size: 9885

metrics.json

@@ -1,4 +1,4 @@
 {
-    "f1": 0.5266106442577031,
-    "roc_auc": 0.8418821541516269
+    "f1": 0.5388888888888889,
+    "roc_auc": 0.847107438016529
 }

params.yaml

@@ -26,4 +26,4 @@ train:
     model_path: models/clf-model.joblib
     params:
         n_estimators: 200
-        max_depth: 20
+        max_depth: 21

reports/feat_imp.csv

@@ -1,9 +1,9 @@
 feature,importance
-num__NumOfProducts,0.20
-num__Age,0.19
-num__IsActiveMember,0.09
-num__Balance,0.02
-num__CreditScore,0.01
+num__NumOfProducts,0.21
+num__Age,0.21
+num__IsActiveMember,0.11
+num__Balance,0.03
+num__CreditScore,0.02
+num__Tenure,0.02
+num__EstimatedSalary,0.01
 num__HasCrCard,0.01
-num__Tenure,0.00
-num__EstimatedSalary,-0.01