dvc.lock

schema: '2.0'
stages:
  data_ingestion:
    cmd: python src/CNN_Classifier/pipeline/stage_01_data_ingestion.py
    deps:
    - path: config/config.yaml
      hash: md5
      md5: 505c1e5080c7f21a3db410c5bb0f88a3
      size: 568
    - path: src/CNN_Classifier/pipeline/stage_01_data_ingestion.py
      hash: md5
      md5: 4bd10e621d7fcb98f972d50513cf7577
      size: 882
    outs:
    - path: artifacts/data_ingestion/kidney-data
      hash: md5
      md5: 29d4b8fc20e111d5418eeda28e3fdd37.dir
      size: 171016859
      nfiles: 1201
  prepare_base_model:
    cmd: python src/CNN_Classifier/pipeline/stage_02_prepare_base_model.py
    deps:
    - path: config/config.yaml
      hash: md5
      md5: 505c1e5080c7f21a3db410c5bb0f88a3
      size: 568
    - path: src/CNN_Classifier/pipeline/stage_02_prepare_base_model.py
      hash: md5
      md5: 8f8ecc694ffff34455f0a5527c619d3b
      size: 912
    params:
      params.yaml:
        CLASSES: 4
        IMAGE_SIZE:
        - 224
        - 224
        - 3
        INCLUDE_TOP: false
        LEARNING_RATE: 0.001
        WEIGHTS: imagenet
    outs:
    - path: artifacts/prepare_base_model
      hash: md5
      md5: 6f8c94b1e425e1fc7a5202b614a7f5e9.dir
      size: 118257246
      nfiles: 2
  training:
    cmd: python src/CNN_Classifier/pipeline/stage_03_model_training.py
    deps:
    - path: artifacts/data_ingestion/kidney-data
      hash: md5
      md5: 29d4b8fc20e111d5418eeda28e3fdd37.dir
      size: 171016859
      nfiles: 1201
    - path: artifacts/prepare_base_model
      hash: md5
      md5: 6f8c94b1e425e1fc7a5202b614a7f5e9.dir
      size: 118257246
      nfiles: 2
    - path: config/config.yaml
      hash: md5
      md5: 505c1e5080c7f21a3db410c5bb0f88a3
      size: 568
    - path: src/CNN_Classifier/pipeline/stage_03_model_training.py
      hash: md5
      md5: 098723b5307eee3e8ec338ca862d5da7
      size: 838
    params:
      params.yaml:
        AUGMENTATION: false
        BATCH_SIZE: 32
        EPOCHS: 1
        IMAGE_SIZE:
        - 224
        - 224
        - 3
    outs:
    - path: artifacts/training/model.keras
      hash: md5
      md5: 934d3e405115c3f1d202681e8ab3cf66
      size: 59736021
  evaluation:
    cmd: python src/CNN_Classifier/pipeline/stage_04_model_evaluation.py
    deps:
    - path: artifacts/data_ingestion/kidney-data
      hash: md5
      md5: 29d4b8fc20e111d5418eeda28e3fdd37.dir
      size: 171016859
      nfiles: 1201
    - path: artifacts/training/model.keras
      hash: md5
      md5: 934d3e405115c3f1d202681e8ab3cf66
      size: 59736021
    - path: config/config.yaml
      hash: md5
      md5: 505c1e5080c7f21a3db410c5bb0f88a3
      size: 568
    - path: src/CNN_Classifier/pipeline/stage_04_model_evaluation.py
      hash: md5
      md5: 9a80492fd45c3a97bc1731c875f56466
      size: 1079
    params:
      params.yaml:
        BATCH_SIZE: 32
        IMAGE_SIZE:
        - 224
        - 224
        - 3
    outs:
    - path: scores.json
      hash: md5
      md5: c3d8f8a15cf31bce1e464ab3a668ac72
      size: 73