dvc.lock

schema: '2.0'
stages:
  data_ingestion:
    cmd: "set PYTHONPATH=%PYTHONPATH%;d:\\\\Kidney-Disease-Classification\\\\src &&
      python src/cnnClassifier/pipeline/stage_01_data_ingestion.py\n"
    deps:
    - path: config/config.yaml
      hash: md5
      md5: 103a88ea8aa06f6e0757ba29f2bf14c3
      size: 582
    - path: src/cnnClassifier/pipeline/stage_01_data_ingestion.py
      hash: md5
      md5: caf501a7205ba64b6d4426666570432c
      size: 908
    outs:
    - path: artifacts/data_ingestion/kidney-ct-scan-image
      hash: md5
      md5: 33ed59dbe5dec8ce2bb8e489b55203e4.dir
      size: 58936381
      nfiles: 465
  prepare_base_model:
    cmd: "set PYTHONPATH=%PYTHONPATH%;d:\\\\Kidney-Disease-Classification\\\\src &&
      python src/cnnClassifier/pipeline/stage_02_prepare_base_model.py\n"
    deps:
    - path: config/config.yaml
      hash: md5
      md5: 103a88ea8aa06f6e0757ba29f2bf14c3
      size: 582
    - 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.01
        WEIGHTS: imagenet
    outs:
    - path: artifacts/prepare_base_model
      hash: md5
      md5: e7c01fc8cc3cb289f83b63b00dbd1c3e.dir
      size: 118054560
      nfiles: 2
  training:
    cmd: "set PYTHONPATH=%PYTHONPATH%;d:\\\\Kidney-Disease-Classification\\\\src &&
      python src/cnnClassifier/pipeline/stage_03_model_training.py\n"
    deps:
    - path: artifacts/data_ingestion/kidney-ct-scan-image
      hash: md5
      md5: 33ed59dbe5dec8ce2bb8e489b55203e4.dir
      size: 58936381
      nfiles: 465
    - path: artifacts/prepare_base_model
      hash: md5
      md5: e7c01fc8cc3cb289f83b63b00dbd1c3e.dir
      size: 118054560
      nfiles: 2
    - path: config/config.yaml
      hash: md5
      md5: 103a88ea8aa06f6e0757ba29f2bf14c3
      size: 582
    - path: src/cnnClassifier/pipeline/stage_03_model_training.py
      hash: md5
      md5: 0b5d68d520496bb7c1c8f5b3d7888dde
      size: 927
    params:
      params.yaml:
        AUGMENTATION: true
        BATCH_SIZE: 16
        EPOCHS: 2
        IMAGE_SIZE:
        - 224
        - 224
        - 3
    outs:
    - path: artifacts/training/model.h5
      hash: md5
      md5: dfcad081b5aa84ddc7b571d69c50ddee
      size: 59337520
  evaluation:
    cmd: "set PYTHONPATH=%PYTHONPATH%;d:\\\\Kidney-Disease-Classification\\\\src &&
      python src/cnnClassifier/pipeline/stage_04_model_evaluation.py\n"
    deps:
    - path: artifacts/data_ingestion/kidney-ct-scan-image
      hash: md5
      md5: 33ed59dbe5dec8ce2bb8e489b55203e4.dir
      size: 58936381
      nfiles: 465
    - path: artifacts/training/model.h5
      hash: md5
      md5: dfcad081b5aa84ddc7b571d69c50ddee
      size: 59337520
    - path: config/config.yaml
      hash: md5
      md5: 103a88ea8aa06f6e0757ba29f2bf14c3
      size: 582
    - path: src/cnnClassifier/pipeline/stage_04_model_evaluation.py
      hash: md5
      md5: e5acfe1c4e9e0a3f116bebf9898cb1ee
      size: 938
    params:
      params.yaml:
        BATCH_SIZE: 16
        IMAGE_SIZE:
        - 224
        - 224
        - 3
    outs:
    - path: scores.json
      hash: md5
      md5: 6b5d419e5e1e1ab85193b96d1c509f01
      size: 72