dvc.lock

schema: '2.0'
stages:
  data_ingestion:
    cmd: python src/cnnClassifier/pipeline/stage_01_data_ingestion.py
    deps:
    - path: config/config.yaml
      hash: md5
      md5: d6a59d48c7f5c604619571b20887eff0
      size: 816
    - path: src/cnnClassifier/pipeline/stage_01_data_ingestion.py
      hash: md5
      md5: b417750f3be47e8e51c11ad2f260a905
      size: 903
    outs:
    - path: artifacts/data_ingestion/Chicken-fecal-images
      hash: md5
      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
      hash: md5
      md5: d6a59d48c7f5c604619571b20887eff0
      size: 816
    - path: src/cnnClassifier/pipeline/stage_02_prepare_base_model.py
      hash: md5
      md5: dd6f32a04e70093fcc4073cb138cdef2
      size: 997
    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: c4ffe3fa1f302f336bc8e4860b8195f9.dir
      size: 118059248
      nfiles: 2
  training:
    cmd: python src/cnnClassifier/pipeline/stage_03_training.py
    deps:
    - path: artifacts/data_ingestion/Chicken-fecal-images
      hash: md5
      md5: 9e1f8dd2eae3c29e9d635df89d438ae4.dir
      size: 12207807
      nfiles: 390
    - path: artifacts/prepare_base_model
      hash: md5
      md5: c4ffe3fa1f302f336bc8e4860b8195f9.dir
      size: 118059248
      nfiles: 2
    - path: config/config.yaml
      hash: md5
      md5: d6a59d48c7f5c604619571b20887eff0
      size: 816
    - path: src/cnnClassifier/components/prepare_callbacks.py
      hash: md5
      md5: df047e424075dcab9aec144951712130
      size: 1006
    - path: src/cnnClassifier/pipeline/stage_03_training.py
      hash: md5
      md5: e282c6aa149ee05234fa987f22537d15
      size: 1266
    params:
      params.yaml:
        AUGMENTATION: true
        BATCH_SIZE: 16
        EPOCHS: 1
        IMAGE_SIZE:
        - 224
        - 224
        - 3
    outs:
    - path: artifacts/training/model.h5
      hash: md5
      md5: d8563589464df59d87c933e1e1818f02
      size: 59340720
  evaluation:
    cmd: python src/cnnClassifier/pipeline/stage_04_evaluation.py
    deps:
    - path: artifacts/data_ingestion/Chicken-fecal-images
      hash: md5
      md5: 9e1f8dd2eae3c29e9d635df89d438ae4.dir
      size: 12207807
      nfiles: 390
    - path: artifacts/training/model.h5
      hash: md5
      md5: d8563589464df59d87c933e1e1818f02
      size: 59340720
    - path: config/config.yaml
      hash: md5
      md5: d6a59d48c7f5c604619571b20887eff0
      size: 816
    - path: src/cnnClassifier/pipeline/stage_04_evaluation.py
      hash: md5
      md5: 8204783faf219204568b3e3793e2e214
      size: 869
    params:
      params.yaml:
        BATCH_SIZE: 16
        IMAGE_SIZE:
        - 224
        - 224
        - 3
    outs:
    - path: scores.json
      hash: md5
      md5: 9dd074ecdf8c36ccc40b465eee14d784
      size: 58