File size: 3,031 Bytes
7195b15
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
schema: '2.0'
stages:
  data_ingestion:
    cmd: python src/cnnClassifier/pipeline/stage_01_data_ingestion.py
    deps:
    - path: config/config.yaml
      hash: md5
      md5: 18c5d166940398f449d80f3bf7ceba78
      size: 601
    - path: src/cnnClassifier/pipeline/stage_01_data_ingestion.py
      hash: md5
      md5: 9ab8c5d8d045a810fdc294c23dba44a2
      size: 906
    outs:
    - path: artifacts/data_ingestion/unzip/kidney-ct-scan-image
      hash: md5
      md5: 33ed59dbe5dec8ce2bb8e489b55203e4.dir
      size: 58936381
      nfiles: 465
  prepare_base_model:
    cmd: python src/cnnClassifier/pipeline/stage_02_prepare_base_model.py
    deps:
    - path: config/config.yaml
      hash: md5
      md5: 18c5d166940398f449d80f3bf7ceba78
      size: 601
    - path: src/cnnClassifier/pipeline/stage_02_prepare_base_model.py
      hash: md5
      md5: e8e39a301f4b90d1b4f2c86acc66ef32
      size: 999
    params:
      params.yaml:
        CLASSES: 2
        IMAGE_SIZE:
        - 224
        - 224
        - 3
        INCLUDE_TOP: false
        LEARNING_RATE: 0.009
        WEIGHTS: imagenet
    outs:
    - path: artifacts/prepare_base_model
      hash: md5
      md5: 186cffa6925a8727cbd781402a5b2d91.dir
      size: 118054560
      nfiles: 2
  training:
    cmd: python src/cnnClassifier/pipeline/stage_03_train_model.py
    deps:
    - path: artifacts/data_ingestion/unzip/kidney-ct-scan-image
      hash: md5
      md5: 33ed59dbe5dec8ce2bb8e489b55203e4.dir
      size: 58936381
      nfiles: 465
    - path: artifacts/prepare_base_model
      hash: md5
      md5: 186cffa6925a8727cbd781402a5b2d91.dir
      size: 118054560
      nfiles: 2
    - path: config/config.yaml
      hash: md5
      md5: 18c5d166940398f449d80f3bf7ceba78
      size: 601
    - path: src/cnnClassifier/pipeline/stage_03_train_model.py
      hash: md5
      md5: 3ef39a6e5a0d665c7c48877e098f3c82
      size: 919
    params:
      params.yaml:
        AUGMENTATION: true
        BATCH_SIZE: 32
        EPOCHS: 2
        IMAGE_SIZE:
        - 224
        - 224
        - 3
    outs:
    - path: artifacts/training/model.h5
      hash: md5
      md5: b860a0e1daa2296bf1ab06265a233dae
      size: 59337520
  evaluation:
    cmd: python src/cnnClassifier/pipeline/stage_04_evaluation.py
    deps:
    - path: artifacts/data_ingestion/unzip/kidney-ct-scan-image
      hash: md5
      md5: 33ed59dbe5dec8ce2bb8e489b55203e4.dir
      size: 58936381
      nfiles: 465
    - path: artifacts/training/model.h5
      hash: md5
      md5: b860a0e1daa2296bf1ab06265a233dae
      size: 59337520
    - path: config/config.yaml
      hash: md5
      md5: 18c5d166940398f449d80f3bf7ceba78
      size: 601
    - path: src/cnnClassifier/pipeline/stage_04_evaluation.py
      hash: md5
      md5: ef1e7f821e740d1e4a9d51a4bf724e68
      size: 888
    params:
      params.yaml:
        BATCH_SIZE: 32
        IMAGE_SIZE:
        - 224
        - 224
        - 3
    outs:
    - path: scores.json
      hash: md5
      md5: 8930e55b40b3d3c2866622648a461a1c
      size: 72