dvc.lock

schema: '2.0'
stages:
  data_ingestion:
    cmd: python src/ChestCancerClassifier/pipeline/stage_01_data_ingestion.py
    deps:
    - path: config/config.yaml
      hash: md5
      md5: 0c395b5a3f1527dab179a4c2cd1b8708
      size: 608
    - path: src/ChestCancerClassifier/pipeline/stage_01_data_ingestion.py
      hash: md5
      md5: 7356264d210203d6aaa3bffa7649f15d
      size: 964
    outs:
    - path: artifacts/data_ingestion/Chest-CT-Scan-data
      hash: md5
      md5: 904fa45d934ce879b3b1933dca6cb2f1.dir
      size: 49247431
      nfiles: 343
  prepare_base_model:
    cmd: python src/ChestCancerClassifier/pipeline/stage_02_prepare_base_model.py
    deps:
    - path: config/config.yaml
      hash: md5
      md5: 0c395b5a3f1527dab179a4c2cd1b8708
      size: 608
    - path: src/ChestCancerClassifier/pipeline/stage_02_prepare_base_model.py
      hash: md5
      md5: 21d947a171823dd67c05025f88218859
      size: 1021
    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: e5bd0bb98d5a30cee1c4698052beca57.dir
      size: 118054560
      nfiles: 2
  training:
    cmd: python src/ChestCancerClassifier/pipeline/stage_03_model_trainer.py
    deps:
    - path: artifacts/data_ingestion/Chest-CT-Scan-data
      hash: md5
      md5: 904fa45d934ce879b3b1933dca6cb2f1.dir
      size: 49247431
      nfiles: 343
    - path: artifacts/prepare_base_model
      hash: md5
      md5: e5bd0bb98d5a30cee1c4698052beca57.dir
      size: 118054560
      nfiles: 2
    - path: config/config.yaml
      hash: md5
      md5: 0c395b5a3f1527dab179a4c2cd1b8708
      size: 608
    - path: src/ChestCancerClassifier/pipeline/stage_03_model_trainer.py
      hash: md5
      md5: b1b182ce5997cd2cb51e3523d59bcce0
      size: 938
    params:
      params.yaml:
        AUGMENTATION: true
        BATCH_SIZE: 16
        EPOCHS: 2
        IMAGE_SIZE:
        - 224
        - 224
        - 3
    outs:
    - path: artifacts/training/model.h5
      hash: md5
      md5: caf8d6f5c1226e24d67bd818a45504cc
      size: 59337520
  evaluation:
    cmd: python src/ChestCancerClassifier/pipeline/stage_04_model_evaluation.py
    deps:
    - path: artifacts/data_ingestion/Chest-CT-Scan-data
      hash: md5
      md5: 904fa45d934ce879b3b1933dca6cb2f1.dir
      size: 49247431
      nfiles: 343
    - path: artifacts/training/model.h5
      hash: md5
      md5: caf8d6f5c1226e24d67bd818a45504cc
      size: 59337520
    - path: config/config.yaml
      hash: md5
      md5: 0c395b5a3f1527dab179a4c2cd1b8708
      size: 608
    - path: src/ChestCancerClassifier/pipeline/stage_04_model_evaluation.py
      hash: md5
      md5: 48bb4f68a0381fa83ae13cade4033e41
      size: 962
    params:
      params.yaml:
        BATCH_SIZE: 16
        IMAGE_SIZE:
        - 224
        - 224
        - 3
    outs:
    - path: scores.json
      hash: md5
      md5: ca2c42f47404a6ff62954da53e97655b
      size: 72