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