dvc.lock

schema: '2.0'
stages:
  data_ingestion:
    cmd: python src/chest_cancer_classifier/pipeline/stage_1_data_ingestion.py
    deps:
    - path: config/config.yaml
      hash: md5
      md5: c34b1a3d1078088dcb9f483aa5971ec5
      size: 1443
    - path: src/chest_cancer_classifier/pipeline/stage_1_data_ingestion.py
      hash: md5
      md5: c86ebcac1360545068f5aadbaf4f776e
      size: 1882
    outs:
    - path: artifacts/data_ingestion/Data
      hash: md5
      md5: 3ddc2c0e262e947578d73eeec3119af6.dir
      size: 124956798
      nfiles: 1000
  prepare_base_model:
    cmd: python src/chest_cancer_classifier/pipeline/stage_2_prepare_base_model.py
    deps:
    - path: config/config.yaml
      hash: md5
      md5: c34b1a3d1078088dcb9f483aa5971ec5
      size: 1443
    - path: src/chest_cancer_classifier/pipeline/stage_2_prepare_base_model.py
      hash: md5
      md5: 48cb79ddc79aa7ce7574ea6d66832c48
      size: 1935
    params:
      params.yaml:
        CLASSES: 4
        IMAGE_SIZE:
        - 224
        - 224
        - 3
        INCLUDE_TOP: false
        LEARNING_RATE: 0.01
        WEIGHTS: imagenet
    outs:
    - path: artifacts/prepare_base_model
      hash: md5
      md5: 093d170faa2be4dcd1f5b855e63f125d.dir
      size: 118274168
      nfiles: 2
  training:
    cmd: python src/chest_cancer_classifier/pipeline/stage_3_model_training.py
    deps:
    - path: artifacts/data_ingestion/Data
      hash: md5
      md5: 3ddc2c0e262e947578d73eeec3119af6.dir
      size: 124956798
      nfiles: 1000
    - path: artifacts/prepare_base_model
      hash: md5
      md5: 093d170faa2be4dcd1f5b855e63f125d.dir
      size: 118274168
      nfiles: 2
    - path: config/config.yaml
      hash: md5
      md5: c34b1a3d1078088dcb9f483aa5971ec5
      size: 1443
    - path: src/chest_cancer_classifier/pipeline/stage_3_model_training.py
      hash: md5
      md5: 7affe72d4439d7780014e9844c727727
      size: 1863
    params:
      params.yaml:
        AUGMENTATION: true
        BATCH_SIZE: 16
        EPOCHS: 5
        IMAGE_SIZE:
        - 224
        - 224
        - 3
    outs:
    - path: artifacts/training/model.h5
      hash: md5
      md5: 39925b7fd1a4a6ec85782fb9545927f3
      size: 60153320
  evaluation:
    cmd: python src/chest_cancer_classifier/pipeline/stage_4_model_evaluation.py
    deps:
    - path: artifacts/data_ingestion/Data
      hash: md5
      md5: 3ddc2c0e262e947578d73eeec3119af6.dir
      size: 124956798
      nfiles: 1000
    - path: artifacts/training/model.h5
      hash: md5
      md5: 39925b7fd1a4a6ec85782fb9545927f3
      size: 60153320
    - path: config/config.yaml
      hash: md5
      md5: c34b1a3d1078088dcb9f483aa5971ec5
      size: 1443
    - path: src/chest_cancer_classifier/pipeline/stage_4_model_evaluation.py
      hash: md5
      md5: 3d60303f5fed6f98b3b16b3fcd8b33fd
      size: 1839
    params:
      params.yaml:
        BATCH_SIZE: 16
        IMAGE_SIZE:
        - 224
        - 224
        - 3
    outs:
    - path: scores.json
      hash: md5
      md5: 3e53d3f918e0950a632a92db7ea4eb02
      size: 73