| schema: '2.0' | |
| stages: | |
| data_ingestion: | |
| cmd: python src/cnnClassifier/pipeline/stage_01_data_ingestion.py | |
| deps: | |
| - path: config/config.yaml | |
| hash: md5 | |
| md5: 18c5d166940398f449d80f3bf7ceba78 | |
| size: 601 | |
| - path: src/cnnClassifier/pipeline/stage_01_data_ingestion.py | |
| hash: md5 | |
| md5: 9ab8c5d8d045a810fdc294c23dba44a2 | |
| size: 906 | |
| outs: | |
| - path: artifacts/data_ingestion/unzip/kidney-ct-scan-image | |
| hash: md5 | |
| md5: 33ed59dbe5dec8ce2bb8e489b55203e4.dir | |
| size: 58936381 | |
| nfiles: 465 | |
| prepare_base_model: | |
| cmd: python src/cnnClassifier/pipeline/stage_02_prepare_base_model.py | |
| deps: | |
| - path: config/config.yaml | |
| hash: md5 | |
| md5: 18c5d166940398f449d80f3bf7ceba78 | |
| size: 601 | |
| - path: src/cnnClassifier/pipeline/stage_02_prepare_base_model.py | |
| hash: md5 | |
| md5: e8e39a301f4b90d1b4f2c86acc66ef32 | |
| size: 999 | |
| params: | |
| params.yaml: | |
| CLASSES: 2 | |
| IMAGE_SIZE: | |
| - 224 | |
| - 224 | |
| - 3 | |
| INCLUDE_TOP: false | |
| LEARNING_RATE: 0.009 | |
| WEIGHTS: imagenet | |
| outs: | |
| - path: artifacts/prepare_base_model | |
| hash: md5 | |
| md5: 186cffa6925a8727cbd781402a5b2d91.dir | |
| size: 118054560 | |
| nfiles: 2 | |
| training: | |
| cmd: python src/cnnClassifier/pipeline/stage_03_train_model.py | |
| deps: | |
| - path: artifacts/data_ingestion/unzip/kidney-ct-scan-image | |
| hash: md5 | |
| md5: 33ed59dbe5dec8ce2bb8e489b55203e4.dir | |
| size: 58936381 | |
| nfiles: 465 | |
| - path: artifacts/prepare_base_model | |
| hash: md5 | |
| md5: 186cffa6925a8727cbd781402a5b2d91.dir | |
| size: 118054560 | |
| nfiles: 2 | |
| - path: config/config.yaml | |
| hash: md5 | |
| md5: 18c5d166940398f449d80f3bf7ceba78 | |
| size: 601 | |
| - path: src/cnnClassifier/pipeline/stage_03_train_model.py | |
| hash: md5 | |
| md5: 3ef39a6e5a0d665c7c48877e098f3c82 | |
| size: 919 | |
| params: | |
| params.yaml: | |
| AUGMENTATION: true | |
| BATCH_SIZE: 32 | |
| EPOCHS: 2 | |
| IMAGE_SIZE: | |
| - 224 | |
| - 224 | |
| - 3 | |
| outs: | |
| - path: artifacts/training/model.h5 | |
| hash: md5 | |
| md5: b860a0e1daa2296bf1ab06265a233dae | |
| size: 59337520 | |
| evaluation: | |
| cmd: python src/cnnClassifier/pipeline/stage_04_evaluation.py | |
| deps: | |
| - path: artifacts/data_ingestion/unzip/kidney-ct-scan-image | |
| hash: md5 | |
| md5: 33ed59dbe5dec8ce2bb8e489b55203e4.dir | |
| size: 58936381 | |
| nfiles: 465 | |
| - path: artifacts/training/model.h5 | |
| hash: md5 | |
| md5: b860a0e1daa2296bf1ab06265a233dae | |
| size: 59337520 | |
| - path: config/config.yaml | |
| hash: md5 | |
| md5: 18c5d166940398f449d80f3bf7ceba78 | |
| size: 601 | |
| - path: src/cnnClassifier/pipeline/stage_04_evaluation.py | |
| hash: md5 | |
| md5: ef1e7f821e740d1e4a9d51a4bf724e68 | |
| size: 888 | |
| params: | |
| params.yaml: | |
| BATCH_SIZE: 32 | |
| IMAGE_SIZE: | |
| - 224 | |
| - 224 | |
| - 3 | |
| outs: | |
| - path: scores.json | |
| hash: md5 | |
| md5: 8930e55b40b3d3c2866622648a461a1c | |
| size: 72 | |