-
Notifications
You must be signed in to change notification settings - Fork 0
/
dvc.lock
117 lines (117 loc) · 2.97 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
schema: '2.0'
stages:
data_ingestion:
cmd: python src/cnnClassifier/pipeline/stage_01.py
deps:
- path: config/config.yaml
hash: md5
md5: 1bb0eeda960aa3d30e5cb4b2ac97048c
size: 586
- path: src/cnnClassifier/pipeline/stage_01.py
hash: md5
md5: caf501a7205ba64b6d4426666570432c
size: 908
outs:
- path: artifacts/data_ingestion/kidney-scan-images
hash: md5
md5: 37515bb11b289cf75a2f07c53c8cceb9.dir
size: 961115981
nfiles: 7360
prepare_base_model:
cmd: python src/cnnClassifier/pipeline/stage_02.py
deps:
- path: config/config.yaml
hash: md5
md5: 1bb0eeda960aa3d30e5cb4b2ac97048c
size: 586
- path: src/cnnClassifier/pipeline/stage_02.py
hash: md5
md5: aa1673f7e6fcc3c4fd16c3dfc7c7aa43
size: 989
- path: src\cnnClassifier\components\prepare_base_model.py
hash: md5
md5: 5733ffc1cef81272f636ad0b414b6790
size: 2227
params:
params.yaml:
CLASSES: 2
IMAGE_SIZE:
- 224
- 224
- 3
INCLUDE_TOP: false
LEARNING_RATE: 0.01
WEIGHTS: imagenet
outs:
- path: artifacts/prepare_base_model
hash: md5
md5: fd4a6b2ac3a3c24bb472a89c8745198d.dir
size: 118054560
nfiles: 2
training:
cmd: python src/cnnClassifier/pipeline/stage_03.py
deps:
- path: artifacts/data_ingestion/kidney-scan-images
hash: md5
md5: 37515bb11b289cf75a2f07c53c8cceb9.dir
size: 961115981
nfiles: 7360
- path: artifacts/prepare_base_model
hash: md5
md5: fd4a6b2ac3a3c24bb472a89c8745198d.dir
size: 118054560
nfiles: 2
- path: config/config.yaml
hash: md5
md5: 1bb0eeda960aa3d30e5cb4b2ac97048c
size: 586
- path: src/cnnClassifier/pipeline/stage_03.py
hash: md5
md5: 5e6448878bfb4cfe6a41e7e4a04ed913
size: 917
params:
params.yaml:
AUGMENTATION: true
BATCH_SIZE: 16
EPOCHS: 1
IMAGE_SIZE:
- 224
- 224
- 3
outs:
- path: artifacts/training/model.h5
hash: md5
md5: 7959caa43e4eaf7a1f900fb6f4c8449e
size: 59337520
evaluation:
cmd: python src/cnnClassifier/pipeline/stage_04.py
deps:
- path: artifacts/data_ingestion/kidney-scan-images
hash: md5
md5: 37515bb11b289cf75a2f07c53c8cceb9.dir
size: 961115981
nfiles: 7360
- path: artifacts/training/model.h5
hash: md5
md5: 7959caa43e4eaf7a1f900fb6f4c8449e
size: 59337520
- path: config/config.yaml
hash: md5
md5: 1bb0eeda960aa3d30e5cb4b2ac97048c
size: 586
- path: src/cnnClassifier/pipeline/stage_04.py
hash: md5
md5: d80d7050afc62570716864422964f73e
size: 932
params:
params.yaml:
BATCH_SIZE: 16
IMAGE_SIZE:
- 224
- 224
- 3
outs:
- path: scores.json
hash: md5
md5: 5b5ee80bc6c28aaad98284cddd54ab24
size: 72