dvc.lock

schema: '2.0'
stages:
  data_ingestion:
    cmd: python src/renalClassifier/pipeline/stage_01_data_ingestion.py
    deps:
    - path: config/config.yaml
      hash: md5
      md5: 3ad8d869d54f1ae69f253a15de9ac4ca
      size: 563
    - path: src/renalClassifier/pipeline/stage_01_data_ingestion.py
      hash: md5
      md5: 9164c60fe70a0a88b43e43e18affbe6a
      size: 942
    outs:
    - path: artifacts/data_ingestion/CT-KIDNEY-DATASET-Normal-Cyst-Tumor-Stone
      hash: md5
      md5: 480f02db6967e2017d804fca0430c1a4.dir
      size: 1661580918
      nfiles: 12446
  prepare_base_model:
    cmd: python src/renalClassifier/pipeline/stage_02_prepare_base_model.py
    deps:
    - path: config/config.yaml
      hash: md5
      md5: 3ad8d869d54f1ae69f253a15de9ac4ca
      size: 563
    - path: src/renalClassifier/pipeline/stage_02_prepare_base_model.py
      hash: md5
      md5: 7e947c2bf52ab9ba385bb41e90ca392f
      size: 981
    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: 38a4d0e49b70f31830ad6d5cc049df4b.dir
      size: 118258808
      nfiles: 2
  training:
    cmd: python src/renalClassifier/pipeline/stage_03_model_training.py
    deps:
    - path: artifacts/data_ingestion/CT-KIDNEY-DATASET-Normal-Cyst-Tumor-Stone
      hash: md5
      md5: 480f02db6967e2017d804fca0430c1a4.dir
      size: 1661580918
      nfiles: 12446
    - path: artifacts/prepare_base_model
      hash: md5
      md5: 38a4d0e49b70f31830ad6d5cc049df4b.dir
      size: 118258808
      nfiles: 2
    - path: config/config.yaml
      hash: md5
      md5: 3ad8d869d54f1ae69f253a15de9ac4ca
      size: 563
    - path: src/renalClassifier/pipeline/stage_03_model_training.py
      hash: md5
      md5: 93c6cd17b423b9246fdb5cf3462d386b
      size: 897
    params:
      params.yaml:
        AUGMENTATION: true
        BATCH_SIZE: 16
        EPOCHS: 5
        IMAGE_SIZE:
        - 224
        - 224
        - 3
    outs:
    - path: artifacts/training/model.h5
      hash: md5
      md5: b16701abfe74a48f678fc6ac4597d962
      size: 59339072
  evaluation:
    cmd: python src/renalClassifier/pipeline/stage_04_model_evaluation.py
    deps:
    - path: artifacts/data_ingestion/CT-KIDNEY-DATASET-Normal-Cyst-Tumor-Stone
      hash: md5
      md5: 480f02db6967e2017d804fca0430c1a4.dir
      size: 1661580918
      nfiles: 12446
    - path: artifacts/training/model.h5
      hash: md5
      md5: b16701abfe74a48f678fc6ac4597d962
      size: 59339072
    - path: config/config.yaml
      hash: md5
      md5: 3ad8d869d54f1ae69f253a15de9ac4ca
      size: 563
    - path: src/renalClassifier/pipeline/stage_04_model_evaluation.py
      hash: md5
      md5: b961a4546db837b6303da67b2bcb197a
      size: 903
    params:
      params.yaml:
        BATCH_SIZE: 16
        IMAGE_SIZE:
        - 224
        - 224
        - 3
    outs:
    - path: scores.json
      hash: md5
      md5: d130f8103bfefe6f1d8d9f60d7c0058e
      size: 70