forked from iterative/demo-bank-customer-churn
-
Notifications
You must be signed in to change notification settings - Fork 0
/
dvc.lock
151 lines (151 loc) · 3.93 KB
/
dvc.lock
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
schema: '2.0'
stages:
data_split:
cmd: python stages/data_split.py --config=params.yaml
deps:
- path: data/Churn_Modelling.csv
md5: 73fba4fd85b6dbaf95c9d5e3836d3926
size: 684860
- path: stages/data_split.py
md5: 519fbe22a8ce712999f28010f3c56fe6
size: 1606
params:
params.yaml:
base:
project: bank_customer_churn
data_dir: data
data_fname: Churn_Modelling.csv
cat_cols:
- Geography
num_cols:
- CreditScore
- Age
- Tenure
- Balance
- NumOfProducts
- HasCrCard
- IsActiveMember
- EstimatedSalary
targ_col: Exited
random_state: 42
model_dir: .mlem/model
data_split:
test_size: 0.25
outs:
- path: data/X_test.pkl
md5: fcd1b7e28f320159a0eed658ee56ba97
size: 186030
- path: data/X_train.pkl
md5: 7caa1ff6b9fd082639f697e702dacb15
size: 556049
- path: data/y_test.pkl
md5: 7743d704b8b97cc303ceef12b9be6832
size: 60613
- path: data/y_train.pkl
md5: c5e58ac55c3f8077ffac66b0e9b3536a
size: 180622
train:
cmd: python stages/train.py --config=params.yaml
deps:
- path: data/X_train.pkl
md5: 7caa1ff6b9fd082639f697e702dacb15
size: 556049
- path: data/y_train.pkl
md5: c5e58ac55c3f8077ffac66b0e9b3536a
size: 180622
- path: stages/train.py
md5: bc6943a84d970ae52fc7c9968f9605bf
size: 2484
params:
params.yaml:
base:
project: bank_customer_churn
data_dir: data
data_fname: Churn_Modelling.csv
cat_cols:
- Geography
num_cols:
- CreditScore
- Age
- Tenure
- Balance
- NumOfProducts
- HasCrCard
- IsActiveMember
- EstimatedSalary
targ_col: Exited
random_state: 42
model_dir: .mlem/model
train:
model_type: randomforest
params:
n_estimators: 150
max_depth: 15
outs:
- path: .mlem/model/clf-model
md5: 6cdb89703cc613f2346a847301f362f1
size: 16506539
- path: .mlem/model/clf-model.mlem
md5: ee258dddafa171cdb2224c69747823b3
size: 3297
eval:
cmd: python stages/eval.py --config=params.yaml
deps:
- path: .mlem/model/clf-model
md5: 6cdb89703cc613f2346a847301f362f1
size: 16506539
- path: .mlem/model/clf-model.mlem
md5: ee258dddafa171cdb2224c69747823b3
size: 3297
- path: data/X_test.pkl
md5: fcd1b7e28f320159a0eed658ee56ba97
size: 186030
- path: data/y_test.pkl
md5: 7743d704b8b97cc303ceef12b9be6832
size: 60613
- path: stages/eval.py
md5: 01cf10dab98aed930e7bc2eea7439b9d
size: 2692
params:
params.yaml:
base:
project: bank_customer_churn
data_dir: data
data_fname: Churn_Modelling.csv
cat_cols:
- Geography
num_cols:
- CreditScore
- Age
- Tenure
- Balance
- NumOfProducts
- HasCrCard
- IsActiveMember
- EstimatedSalary
targ_col: Exited
random_state: 42
model_dir: .mlem/model
data_split:
test_size: 0.25
train:
model_type: randomforest
params:
n_estimators: 150
max_depth: 15
outs:
- path: eval_plots/cm.png
md5: f11849cd529001eeb85d4ff4b4e6c0be
size: 10231
- path: eval_plots/plots/roc.json
md5: 2a0c382d788fddb082156a4db781b707
size: 79220
- path: feat_imp.csv
md5: cbf7ad5a2bbffe802927b560f317d913
size: 297
- path: metrics.json
md5: 096c20d40fcb880531a5997d423f0fe2
size: 67
- path: models/perm_imp.joblib
md5: 9d70de3ce43e487f6ffded0be02c2265
size: 16499964