Hugging Face's logo

Spaces:
Sentoz
/
kidney-classifier
Sleeping

App Files Community
kidney-classifier
File size: 4,308 Bytes
3e93e14
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
schema: '2.0'
stages:
  data_ingestion:
    cmd: python src/cnnClassifier/pipeline/stage_01_data_ingestion.py
    deps:
    - path: config/config.yaml
      hash: md5
      md5: 20cd3ab789ce919b3687442bc4f2ab85
      size: 1016
    - path: src/cnnClassifier/components/data_ingestion.py
      hash: md5
      md5: f07cc7fed589b1f7d14a637aa94a0433
      size: 1039
    - path: src/cnnClassifier/config/configuration.py
      hash: md5
      md5: d41b12b3ad8d16ad963a9a34118da1ec
      size: 2873
    - path: src/cnnClassifier/pipeline/stage_01_data_ingestion.py
      hash: md5
      md5: e501eeb64cda076b3e15b447e55d6463
      size: 908
    outs:
    - path: artifacts/data_ingestion
      hash: md5
      md5: 86510b1e2ff6da777357ccfdc278e4c8.dir
      size: 116493584
      nfiles: 466
  prepare_base_model:
    cmd: python src/cnnClassifier/pipeline/stage_02_prepare_base_model.py
    deps:
    - path: config/config.yaml
      hash: md5
      md5: 20cd3ab789ce919b3687442bc4f2ab85
      size: 1016
    - path: params.yaml
      hash: md5
      md5: 156a3b540bf80876a34e08b09faaf4fb
      size: 151
    - path: src/cnnClassifier/components/prepare_base_model.py
      hash: md5
      md5: 3c5230f332299193cb460420f3ce5057
      size: 2063
    - path: src/cnnClassifier/config/configuration.py
      hash: md5
      md5: d41b12b3ad8d16ad963a9a34118da1ec
      size: 2873
    - path: src/cnnClassifier/pipeline/stage_02_prepare_base_model.py
      hash: md5
      md5: c886276ae57285dac8969b02bf9077ed
      size: 954
    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: d761158dc61a51df0233a4d98a02499f.dir
      size: 118073624
      nfiles: 2
  training:
    cmd: python src/cnnClassifier/pipeline/stage_03_model_trainer.py
    deps:
    - path: artifacts/data_ingestion/kidney-ct-scan-image
      hash: md5
      md5: 33ed59dbe5dec8ce2bb8e489b55203e4.dir
      size: 58936381
      nfiles: 465
    - path: artifacts/prepare_base_model/base_model_updated.h5
      hash: md5
      md5: 12a1e3ebb90d89346ff2beb4fa21053b
      size: 59147544
    - path: config/config.yaml
      hash: md5
      md5: 20cd3ab789ce919b3687442bc4f2ab85
      size: 1016
    - path: params.yaml
      hash: md5
      md5: 156a3b540bf80876a34e08b09faaf4fb
      size: 151
    - path: src/cnnClassifier/components/model_trainer.py
      hash: md5
      md5: bc19f92e2812f36ba12a7c66730a6e21
      size: 2675
    - path: src/cnnClassifier/config/configuration.py
      hash: md5
      md5: d41b12b3ad8d16ad963a9a34118da1ec
      size: 2873
    - path: src/cnnClassifier/pipeline/stage_03_model_trainer.py
      hash: md5
      md5: 0951b497a475aac360e1c73b347b2295
      size: 885
    params:
      params.yaml:
        AUGMENTATION: true
        BATCH_SIZE: 16
        EPOCHS: 5
        IMAGE_SIZE:
        - 224
        - 224
        - 3
        LEARNING_RATE: 0.01
    outs:
    - path: artifacts/training/model.h5
      hash: md5
      md5: 87e2a46b9573a6bba1da41192f0dff18
      size: 59147544
  evaluation:
    cmd: python src/cnnClassifier/pipeline/stage_04_model_evaluation.py
    deps:
    - path: artifacts/training/model.h5
      hash: md5
      md5: 87e2a46b9573a6bba1da41192f0dff18
      size: 59147544
    - path: config/config.yaml
      hash: md5
      md5: 20cd3ab789ce919b3687442bc4f2ab85
      size: 1016
    - path: src/cnnClassifier/components/model_evaluation_mlflow.py
      hash: md5
      md5: 4612a6a44af8961549348656ece0c848
      size: 2306
    - path: src/cnnClassifier/config/configuration.py
      hash: md5
      md5: d41b12b3ad8d16ad963a9a34118da1ec
      size: 2873
    - path: src/cnnClassifier/pipeline/stage_04_model_evaluation.py
      hash: md5
      md5: dff6dc7d6115804b764c98c53f3bbc43
      size: 908
    params:
      params.yaml:
        BATCH_SIZE: 16
        IMAGE_SIZE:
        - 224
        - 224
        - 3
    outs:
    - path: scores.json
      hash: md5
      md5: 1fad9e68e2b1611a3fc59b064c62106e
      size: 72