schema: '2.0'
stages:
  data_ingestion:
    cmd: python src/kidney_classification/pipeline/stage_01_data_ingestion.py
    deps:
    - path: config/config.yaml
      hash: md5
      md5: 23d61aa500e4da63569da56d61ddb49e
      size: 568
    - path: src/kidney_classification/pipeline/stage_01_data_ingestion.py
      hash: md5
      md5: 0496157b33ff2c7182935a33c605f461
      size: 955
    outs:
    - path: artifacts/data_ingestion/CT-KIDNEY-DATASET-Normal-Cyst-Tumor-Stone
      hash: md5
      md5: 47292b3e13804acbce0f2b4ce55edc57.dir
      size: 887719156
      nfiles: 5511
  prepare_base_model:
    cmd: python src/kidney_classification/pipeline/stage_02_prepare_base_model.py
    deps:
    - path: config/config.yaml
      hash: md5
      md5: 23d61aa500e4da63569da56d61ddb49e
      size: 568
    - path: src/kidney_classification/pipeline/stage_02_prepare_base_model.py
      hash: md5
      md5: 27f16408dee8b42a215b576052e54bb2
      size: 950
    params:
      params.yaml:
        CLASSES: 4
        IMAGE_SIZE:
        - 150
        - 150
        - 3
        INCLUDE_TOP: false
        WEIGHTS: imagenet
    outs:
    - path: artifacts/prepare_base_model
      hash: md5
      md5: 2721c46e50849883acd55e5d4e3dcefb.dir
      size: 60010269
      nfiles: 1
  training:
    cmd: python src/kidney_classification/pipeline/stage_03_model_training.py
    deps:
    - path: artifacts/data_ingestion/CT-KIDNEY-DATASET-Normal-Cyst-Tumor-Stone
      hash: md5
      md5: ec42dfce2ae993cf49f6d499a389c93e.dir
      size: 1661580918
      nfiles: 12446
    - path: artifacts/prepare_base_model
      hash: md5
      md5: a4f718a24c253b4e539f7ba2dc9d3442.dir
      size: 59997688
      nfiles: 1
    - path: config/config.yaml
      hash: md5
      md5: bc47b5f88a0220822ff7921144b69204
      size: 565
    - path: src/kidney_classification/pipeline/stage_03_model_training.py
      hash: md5
      md5: cb8342c5c2f23c4d395f299924161231
      size: 941
    params:
      params.yaml:
        BATCH_SIZE: 32
        EPOCHS: 15
        IMAGE_SIZE:
        - 150
        - 150
        - 3
    outs:
    - path: artifacts/training/model.h5
      hash: md5
      md5: 17969099556c165c584938f53a3fc085
      size: 62136448
  evaluation:
    cmd: python src/kidney_classification/pipeline/stage_04_model_evaluation_with_mlflow.py
    deps:
    - path: artifacts/data_ingestion/CT-KIDNEY-DATASET-Normal-Cyst-Tumor-Stone
      hash: md5
      md5: ec42dfce2ae993cf49f6d499a389c93e.dir
      size: 1661580918
      nfiles: 12446
    - path: artifacts/training/model.h5
      hash: md5
      md5: 17969099556c165c584938f53a3fc085
      size: 62136448
    - path: config/config.yaml
      hash: md5
      md5: bc47b5f88a0220822ff7921144b69204
      size: 565
    - path: src/kidney_classification/pipeline/stage_04_model_evaluation_with_mlflow.py
      hash: md5
      md5: 7c91c2fdf529e4dcf7dec2684eb3d212
      size: 892
    params:
      params.yaml:
        BATCH_SIZE: 32
        EPOCHS: 15
        IMAGE_SIZE:
        - 150
        - 150
        - 3
    outs:
    - path: scores.json
      hash: md5
      md5: e80b69c44a4cf771b208a549d9e5ae30
      size: 58