dvc.lock

schema: '2.0'
stages:
  data_ingestion:
    cmd: python src/cnnClassifier/pipeline/stage_01_data_ingestion.py
    deps:
    - path: config/config.yaml
      md5: 3f6dc733e0735665409e293214355568
      size: 818
    - path: src/cnnClassifier/pipeline/stage_01_data_ingestion.py
      md5: b0ff51aae578695575ba98f83c03d86e
      size: 914
    outs:
    - path: artifacts/data_ingestion/Chicken-fecal-images
      md5: 9e1f8dd2eae3c29e9d635df89d438ae4.dir
      size: 12207807
      nfiles: 390
  prepare_base_model:
    cmd: python src/cnnClassifier/pipeline/stage_02_prepare_base_model.py
    deps:
    - path: config/config.yaml
      md5: 3f6dc733e0735665409e293214355568
      size: 818
    - path: src/cnnClassifier/pipeline/stage_02_prepare_base_model.py
      md5: 5bbb3b3d7f36b28620ae359473f09153
      size: 1003
    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: 5ef67f8c7bf546d4aa5249d5c4b2c42c.dir
      size: 118059248
      nfiles: 2
  training:
    cmd: python src/cnnClassifier/pipeline/stage_03_training.py
    deps:
    - path: artifacts/data_ingestion/Chicken-fecal-images
      md5: 9e1f8dd2eae3c29e9d635df89d438ae4.dir
      size: 12207807
      nfiles: 390
    - path: artifacts/prepare_base_model
      hash: md5
      md5: 5ef67f8c7bf546d4aa5249d5c4b2c42c.dir
      size: 118059248
      nfiles: 2
    - path: config/config.yaml
      md5: 3f6dc733e0735665409e293214355568
      size: 818
    - path: src/cnnClassifier/components/prepare_callbacks.py
      hash: md5
      md5: 10c2b80338d446a9aee8b2e1d7189675
      size: 1308
    - path: src/cnnClassifier/pipeline/stage_03_training.py
      md5: 040b50b39a7a78cd8f367c558ebf6dd1
      size: 1268
    params:
      params.yaml:
        AUGMENTATION: true
        BATCH_SIZE: 16
        EPOCHS: 10
        IMAGE_SIZE:
        - 224
        - 224
        - 3
    outs:
    - path: artifacts/training/model.h5
      hash: md5
      md5: bbfbd59c0a99e4ef6308def685c79887
      size: 59340720
  evaluation:
    cmd: python src/cnnClassifier/pipeline/stage_04_evaluation.py
    deps:
    - path: artifacts/data_ingestion/Chicken-fecal-images
      md5: 9e1f8dd2eae3c29e9d635df89d438ae4.dir
      size: 12207807
      nfiles: 390
    - path: artifacts/training/model.h5
      hash: md5
      md5: bbfbd59c0a99e4ef6308def685c79887
      size: 59340720
    - path: config/config.yaml
      md5: 3f6dc733e0735665409e293214355568
      size: 818
    - path: src/cnnClassifier/pipeline/stage_04_evaluation.py
      md5: 66a2ec96bccfd0f56597462ebd711819
      size: 883
    params:
      params.yaml:
        BATCH_SIZE: 16
        IMAGE_SIZE:
        - 224
        - 224
        - 3
    outs:
    - path: scores.json
      hash: md5
      md5: 329dff3adf251dabbf5e2a5bf9d038c4
      size: 73