-
Notifications
You must be signed in to change notification settings - Fork 0
/
dvc.lock
70 lines (70 loc) · 1.78 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
schema: '2.0'
stages:
prepare:
cmd: python3 data/data_preparation.py
deps:
- path: data/data_preparation.py
md5: edd448fe1419fc5358a8a40f60b73ef7
size: 1463
- path: data/raw_data.csv
md5: 639b6192cb043d171b945265f09bc0ea
size: 4409700
params:
params.yaml:
prepare.test_size: 0.2
prepare.undersampling: 0.9
prepare.val_size: 0.2
outs:
- path: holdout.csv
md5: 695840b2c809eaba8439da5caae1f6ba
size: 918287
- path: train.csv
md5: 19fd50d871e25a49415ce8a08c9e8cde
size: 2645118
- path: val.csv
md5: 2682beac5efd57fc57219aef4316054c
size: 733800
train:
cmd: python3 model/model_training.py
deps:
- path: model/model_training.py
md5: de2a654f4b1b8a03dbf4bb95e1056a0d
size: 2791
- path: train.csv
md5: 19fd50d871e25a49415ce8a08c9e8cde
size: 2645118
- path: val.csv
md5: 2682beac5efd57fc57219aef4316054c
size: 733800
params:
params.yaml:
train.criterion: gini
train.max_depth: 4
train.min_sample_leaf: 10
train.n_estimators: 100
outs:
- path: metrics.json
md5: 4142f72abf1f434733323d538018b015
size: 533
- path: model.pkl
md5: e5c8ca4737fac879fb7ec1743d9a8ef6
size: 182739
test:
cmd: python3 -m pytest
deps:
- path: holdout.csv
md5: 695840b2c809eaba8439da5caae1f6ba
size: 918287
- path: model.pkl
md5: e5c8ca4737fac879fb7ec1743d9a8ef6
size: 182739
- path: test/test_data_and_model.py
md5: d6cf6ec5f2c2c212b7d5b76b377216e0
size: 2644
outs:
- path: prc.json
md5: 45246e4b68b8c69476c7e38d79dd7869
size: 139317
- path: rocauc.json
md5: ab7677663d7ce1a39485438c4596d10c
size: 36