dvc.lock

schema: '2.0'
stages:
  data_ingestion:
    cmd: python -m src.cnnClassifier.pipeline.stage_01_data_ingestion
    deps:
    - path: config/config.yaml
      hash: md5
      md5: 583e372fa36974aebb354c168725b9dd
      size: 568
    - path: src/cnnClassifier/pipeline/stage_01_data_ingestion.py
      hash: md5
      md5: f6a4889b0766ac5d3b66b672eefd12ac
      size: 892
    outs:
    - path: artifacts/data_ingestion/kidney-ct-scan-image
      hash: md5
      md5: d9434313dbc8c18071065d64aaccc639.dir
      size: 880129655
      nfiles: 6454
  prepare_base_model:
    cmd: python -m src.cnnClassifier.pipeline.stage_02_prepare_base_model
    deps:
    - path: config/config.yaml
      hash: md5
      md5: 583e372fa36974aebb354c168725b9dd
      size: 568
    - path: src/cnnClassifier/pipeline/stage_02_prepare_base_model.py
      hash: md5
      md5: 2aa4a233cc57fb0e2d0d85a78c9be478
      size: 976
    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: 25e579f32901bd9ff051f5d4c8d14cbc.dir
      size: 118073464
      nfiles: 2
  training:
    cmd: python -m src.cnnClassifier.pipeline.stage_03_model_training
    deps:
    - path: artifacts/data_ingestion/kidney-ct-scan-image
      hash: md5
      md5: d9434313dbc8c18071065d64aaccc639.dir
      size: 880129655
      nfiles: 6454
    - path: artifacts/prepare_base_model
      hash: md5
      md5: 25e579f32901bd9ff051f5d4c8d14cbc.dir
      size: 118073464
      nfiles: 2
    - path: config/config.yaml
      hash: md5
      md5: 583e372fa36974aebb354c168725b9dd
      size: 568
    - path: src/cnnClassifier/pipeline/stage_03_model_training.py
      hash: md5
      md5: 359cff5e44aed64bd99fcb8b12b9641c
      size: 893
    params:
      params.yaml:
        AUGMENTATION: true
        BATCH_SIZE: 16
        EPOCHS: 1
        IMAGE_SIZE:
        - 224
        - 224
        - 3
        LEARNING_RATE: 0.01
    outs:
    - path: artifacts/training/model.h5
      hash: md5
      md5: 958dd2cc98ec35e4d6ba1210ed251db1
      size: 177289736
  evaluation:
    cmd: python -m src.cnnClassifier.pipeline.stage_04_model_evaluation
    deps:
    - path: artifacts/data_ingestion/kidney-ct-scan-image
      hash: md5
      md5: d9434313dbc8c18071065d64aaccc639.dir
      size: 880129655
      nfiles: 6454
    - path: artifacts/training/model.h5
      hash: md5
      md5: 958dd2cc98ec35e4d6ba1210ed251db1
      size: 177289736
    - path: config/config.yaml
      hash: md5
      md5: 583e372fa36974aebb354c168725b9dd
      size: 568
    - path: src/cnnClassifier/pipeline/stage_04_model_evaluation.py
      hash: md5
      md5: 35bd096895322fa256ec621de4a477d1
      size: 899
    params:
      params.yaml:
        BATCH_SIZE: 16
        IMAGE_SIZE:
        - 224
        - 224
        - 3
    outs:
    - path: scores.json
      hash: md5
      md5: 673775297e1f3f38492ca5047c2c4c26
      size: 69