diff --git a/.github/workflows/train-model.yaml b/.github/workflows/train-model.yaml index a862ff7..83348fe 100644 --- a/.github/workflows/train-model.yaml +++ b/.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 diff --git a/dvc.lock b/dvc.lock index 42756e6..7c7356a 100644 --- a/dvc.lock +++ b/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 diff --git a/metrics.json b/metrics.json index 48c5656..b38cbc9 100644 --- a/metrics.json +++ b/metrics.json @@ -1,4 +1,4 @@ { - "f1": 0.5266106442577031, - "roc_auc": 0.8418821541516269 + "f1": 0.5388888888888889, + "roc_auc": 0.847107438016529 } \ No newline at end of file diff --git a/params.yaml b/params.yaml index 031f93b..1c8141c 100644 --- a/params.yaml +++ b/params.yaml @@ -26,4 +26,4 @@ train: model_path: models/clf-model.joblib params: n_estimators: 200 - max_depth: 20 \ No newline at end of file + max_depth: 21 \ No newline at end of file diff --git a/reports/feat_imp.csv b/reports/feat_imp.csv index 3368b60..85c3dd8 100644 --- a/reports/feat_imp.csv +++ b/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 diff --git a/reports/figures/cm.png b/reports/figures/cm.png index 41a9321..9558abf 100644 Binary files a/reports/figures/cm.png and b/reports/figures/cm.png differ