dvc.lock

schema: '2.0'
stages:
  data_ingestion:
    cmd: python src\lung_cancer_classifier\training_stages\stage_01_data_ingestion.py
    deps:
    - path: config\config.yaml
      hash: md5
      md5: 850bb0930ceb0bc258dd6313ee388dcd
      size: 788
    - path: src\lung_cancer_classifier\training_stages\stage_01_data_ingestion.py
      hash: md5
      md5: 34127337de80c625e7d2f842ff0468a3
      size: 1045
    outs:
    - path: artifacts\data_ingestion\Chest-CT-Scan-data
      hash: md5
      md5: 904fa45d934ce879b3b1933dca6cb2f1.dir
      size: 49247431
      nfiles: 343
  prepare_base_model:
    cmd: python src\lung_cancer_classifier\training_stages\stage_02_prepare_base_model.py
    deps:
    - path: config\config.yaml
      hash: md5
      md5: 850bb0930ceb0bc258dd6313ee388dcd
      size: 788
    - path: params.yaml
      hash: md5
      md5: 0214a968c781c259b718f5c9c62f2fe0
      size: 172
    - path: src\lung_cancer_classifier\training_stages\stage_02_prepare_base_model.py
      hash: md5
      md5: 158383fc3ecce8fa6e646c2d239b3c6e
      size: 1107
    params:
      params.yaml:
        CLASSES: 2
        IMAGE_SIZE:
        - 224
        - 224
        - 3
        INCLUDE_TOP: false
        LEARNING_RATE: 0.001
        WEIGHTS: imagenet
    outs:
    - path: artifacts\prepare_base_model
      hash: md5
      md5: 60abf63437cbb198eb76cf57513cc244.dir
      size: 143527560
      nfiles: 2
  training:
    cmd: python src\lung_cancer_classifier\training_stages\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: 60abf63437cbb198eb76cf57513cc244.dir
      size: 143527560
      nfiles: 2
    - path: config\config.yaml
      hash: md5
      md5: 850bb0930ceb0bc258dd6313ee388dcd
      size: 788
    - path: params.yaml
      hash: md5
      md5: 0214a968c781c259b718f5c9c62f2fe0
      size: 172
    - path: src\lung_cancer_classifier\training_stages\stage_03_model_trainer.py
      hash: md5
      md5: 1e5bb74f4ea9b0aa3166e0d15e2d242d
      size: 1036
    params:
      params.yaml:
        AUGMENTATION: true
        BATCH_SIZE: 14
        EPOCHS: 50
        IMAGE_SIZE:
        - 224
        - 224
        - 3
    outs:
    - path: artifacts\training\trained_model.h5
      hash: md5
      md5: 307614e513cf99474909bb1da632406e
      size: 136003600
  evaluation:
    cmd: python src\lung_cancer_classifier\training_stages\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\trained_model.h5
      hash: md5
      md5: 307614e513cf99474909bb1da632406e
      size: 136003600
    - path: config\config.yaml
      hash: md5
      md5: 850bb0930ceb0bc258dd6313ee388dcd
      size: 788
    - path: params.yaml
      hash: md5
      md5: 0214a968c781c259b718f5c9c62f2fe0
      size: 172
    - path: src\lung_cancer_classifier\training_stages\stage_04_model_evaluation.py
      hash: md5
      md5: e106f69c6d78cb345952e0c9c2be1f3a
      size: 1012
    params:
      params.yaml:
        BATCH_SIZE: 14
        IMAGE_SIZE:
        - 224
        - 224
        - 3
    outs:
    - path: scores.json
      hash: md5
      md5: 841c3e0ad9f85a74955b823656679b5c
      size: 58