update

alex000kim · Aug 21, 2023 · c1b20e3 · c1b20e3 · alex000kim · Aug 21, 2023
1 parent 95c1bb7
commit c1b20e3
Show file tree

Hide file tree

Showing 5 changed files with 20 additions and 20 deletions.
diff --git a/dvc.lock b/dvc.lock
@@ -77,12 +77,12 @@ stages:
           model_path: models/clf-model.joblib
           params:
             n_estimators: 70
-            max_depth: 20
+            max_depth: 25
     outs:
     - path: models/clf-model.joblib
       hash: md5
-      md5: 4a2c6cbc9fa3ad451151707153b23a71
-      size: 737282
+      md5: 03393af585c07f182d7e07ecd917cbf6
+      size: 751835
   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: 4a2c6cbc9fa3ad451151707153b23a71
-      size: 737282
+      md5: 03393af585c07f182d7e07ecd917cbf6
+      size: 751835
     - path: src/stages/eval.py
       hash: md5
       md5: e402ec2f10cedbc6c02331cf13cd1147
@@ -127,17 +127,17 @@ stages:
           model_path: models/clf-model.joblib
           params:
             n_estimators: 70
-            max_depth: 20
+            max_depth: 25
     outs:
     - path: metrics.json
       hash: md5
-      md5: 00c66ff9ee1aafe6d0fba1ebf79eb190
+      md5: b9f2c4c1d2e05a80a41ffbd6153dbc5e
       size: 67
     - path: reports/feat_imp.csv
       hash: md5
-      md5: 3e5fc61b56abcaa2157ba8912d90817f
+      md5: c6053f167552337962d76b72bf96ccea
       size: 187
     - path: reports/figures/cm.png
       hash: md5
-      md5: 142d83bcab0b5c8c47a2844b6fedb989
-      size: 10241
+      md5: de6f975b84d200d8539aaeb56f16241b
+      size: 10348
diff --git a/metrics.json b/metrics.json
@@ -1,4 +1,4 @@
 {
-    "f1": 0.5194805194805194,
-    "roc_auc": 0.8168701552298863
+    "f1": 0.5116279069767441,
+    "roc_auc": 0.8161304956705261
 }
diff --git a/params.yaml b/params.yaml
@@ -26,4 +26,4 @@ train:
   model_path: models/clf-model.joblib
   params:
     n_estimators: 70
-    max_depth: 20
+    max_depth: 25
diff --git a/reports/feat_imp.csv b/reports/feat_imp.csv
@@ -1,9 +1,9 @@
 feature,importance
-num__Age,0.20
-num__NumOfProducts,0.19
-num__IsActiveMember,0.08
-num__Balance,0.05
-num__CreditScore,0.02
+num__Age,0.19
+num__NumOfProducts,0.18
+num__IsActiveMember,0.07
+num__Balance,0.04
 num__HasCrCard,0.01
-num__EstimatedSalary,-0.00
-num__Tenure,-0.00
+num__CreditScore,0.00
+num__Tenure,-0.01
+num__EstimatedSalary,-0.01
diff --git a/reports/figures/cm.png b/reports/figures/cm.png
feature	importance
num__Age	0.19
num__NumOfProducts	0.18
num__IsActiveMember	0.07
num__Balance	0.04
num__HasCrCard	0.01
num__CreditScore	0.00
num__Tenure	-0.01
num__EstimatedSalary	-0.01