-
Notifications
You must be signed in to change notification settings - Fork 0
/
dvc.lock
113 lines (113 loc) · 2.94 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
schema: '2.0'
stages:
data_ingestion:
cmd: python src/skindisease/pipeline/stage_01_data_ingestion.py
deps:
- path: config/config.yaml
hash: md5
md5: 5a244e4df1403b3112aa832a7effe6f7
size: 578
- path: src/skindisease/pipeline/stage_01_data_ingestion.py
hash: md5
md5: 8fca5dfeb69795a6289c8a17791b2eec
size: 906
outs:
- path: artifacts/data_ingestion/skin-classification
hash: md5
md5: 0719e87fa7f1c9956dd99183dc4d92a0.dir
size: 14329506
nfiles: 925
prepare_base_model:
cmd: python src/skindisease/pipeline/stage_02_prepare_base_model.py
deps:
- path: config/config.yaml
hash: md5
md5: 5a244e4df1403b3112aa832a7effe6f7
size: 578
- path: src/skindisease/pipeline/stage_02_prepare_base_model.py
hash: md5
md5: b666c0d3834cf09ce437f69fc606ed61
size: 983
params:
params.yaml:
CLASSES: 8
IMAGE_SIZE:
- 224
- 224
- 3
INCLUDE_TOP: false
LEARNING_RATE: 0.01
WEIGHTS: imagenet
outs:
- path: artifacts/prepare_base_model
hash: md5
md5: f64bb1d0455ef3fc1750ffc13c71771d.dir
size: 118656672
nfiles: 2
training:
cmd: python src/skindisease/pipeline/stage_03_model_trainer.py
deps:
- path: artifacts/data_ingestion/skin-classification
hash: md5
md5: 0719e87fa7f1c9956dd99183dc4d92a0.dir
size: 14329506
nfiles: 925
- path: artifacts/prepare_base_model
hash: md5
md5: f64bb1d0455ef3fc1750ffc13c71771d.dir
size: 118656672
nfiles: 2
- path: config/config.yaml
hash: md5
md5: 5a244e4df1403b3112aa832a7effe6f7
size: 578
- path: src/skindisease/pipeline/stage_03_model_trainer.py
hash: md5
md5: b589ed62963f2ebc290c9bc6d96b1b0b
size: 912
params:
params.yaml:
AUGMENTATION: true
BATCH_SIZE: 16
EPOCHS: 60
IMAGE_SIZE:
- 224
- 224
- 3
outs:
- path: artifacts/training/model.h5
hash: md5
md5: 74439ed9d7c37dba2f73479392f9b324
size: 60541744
evaluation:
cmd: python src/skindisease/pipeline/stage_04_model_evaluation.py
deps:
- path: artifacts/data_ingestion/skin-classification
hash: md5
md5: 0719e87fa7f1c9956dd99183dc4d92a0.dir
size: 14329506
nfiles: 925
- path: artifacts/training/model.h5
hash: md5
md5: 74439ed9d7c37dba2f73479392f9b324
size: 60541744
- path: config/config.yaml
hash: md5
md5: 5a244e4df1403b3112aa832a7effe6f7
size: 578
- path: src/skindisease/pipeline/stage_04_model_evaluation.py
hash: md5
md5: 179a624e58c61f0d175730e425ef1748
size: 928
params:
params.yaml:
BATCH_SIZE: 16
IMAGE_SIZE:
- 224
- 224
- 3
outs:
- path: scores.json
hash: md5
md5: b1474447326cdb1c0643443bde708d15
size: 73