Spaces:
Sleeping
Sleeping
| schema: '2.0' | |
| stages: | |
| data_ingestion: | |
| cmd: python src/cnnClassifier/pipeline/stage_01_data_ingestion.py | |
| deps: | |
| - path: config/config.yaml | |
| hash: md5 | |
| md5: d4c6e6a52ca35ea93094c3e1a421499e | |
| size: 578 | |
| - path: src/cnnClassifier/pipeline/stage_01_data_ingestion.py | |
| hash: md5 | |
| md5: bad788253475f50d44fdaa7237967b49 | |
| size: 883 | |
| outs: | |
| - path: artifacts/data_ingestion/Chest-CT-Scan-data | |
| hash: md5 | |
| md5: 904fa45d934ce879b3b1933dca6cb2f1.dir | |
| size: 49247431 | |
| nfiles: 343 | |
| prepare_base_model: | |
| cmd: python src/cnnClassifier/pipeline/stage_02_prepare_base_model.py | |
| deps: | |
| - path: config/config.yaml | |
| hash: md5 | |
| md5: d4c6e6a52ca35ea93094c3e1a421499e | |
| size: 578 | |
| - path: src/cnnClassifier/pipeline/stage_02_prepare_base_model.py | |
| hash: md5 | |
| md5: 30c63470719d961e32045908b7c0772d | |
| size: 966 | |
| 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: 4aa2611cd37984c188512d3a19c6942b.dir | |
| size: 118054560 | |
| nfiles: 2 | |
| training: | |
| cmd: python src/cnnClassifier/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: 4aa2611cd37984c188512d3a19c6942b.dir | |
| size: 118054560 | |
| nfiles: 2 | |
| - path: config/config.yaml | |
| hash: md5 | |
| md5: d4c6e6a52ca35ea93094c3e1a421499e | |
| size: 578 | |
| - path: src/cnnClassifier/pipeline/stage_03_model_trainer.py | |
| hash: md5 | |
| md5: c33e23d2c123f157b2ab007c8e9d938f | |
| size: 893 | |
| params: | |
| params.yaml: | |
| AUGMENTATION: true | |
| BATCH_SIZE: 16 | |
| EPOCHS: 10 | |
| IMAGE_SIZE: | |
| - 224 | |
| - 224 | |
| - 3 | |
| outs: | |
| - path: artifacts/training/model.h5 | |
| hash: md5 | |
| md5: 233944d4fbed7856cf28be27c602014d | |
| size: 59337520 | |
| evaluation: | |
| cmd: python src/cnnClassifier/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: 233944d4fbed7856cf28be27c602014d | |
| size: 59337520 | |
| - path: config/config.yaml | |
| hash: md5 | |
| md5: d4c6e6a52ca35ea93094c3e1a421499e | |
| size: 578 | |
| - path: src/cnnClassifier/pipeline/stage_04_model_evaluation.py | |
| hash: md5 | |
| md5: d20a1645fd93cae9c7c0cecd8a0d4a2a | |
| size: 1188 | |
| params: | |
| params.yaml: | |
| BATCH_SIZE: 16 | |
| IMAGE_SIZE: | |
| - 224 | |
| - 224 | |
| - 3 | |
| outs: | |
| - path: scores.json | |
| hash: md5 | |
| md5: 2c6b298a9827df6c174dc1bbbf40d040 | |
| size: 59 | |