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: 20cd3ab789ce919b3687442bc4f2ab85 | |
| size: 1016 | |
| - path: src/cnnClassifier/components/data_ingestion.py | |
| hash: md5 | |
| md5: f07cc7fed589b1f7d14a637aa94a0433 | |
| size: 1039 | |
| - path: src/cnnClassifier/config/configuration.py | |
| hash: md5 | |
| md5: d41b12b3ad8d16ad963a9a34118da1ec | |
| size: 2873 | |
| - path: src/cnnClassifier/pipeline/stage_01_data_ingestion.py | |
| hash: md5 | |
| md5: e501eeb64cda076b3e15b447e55d6463 | |
| size: 908 | |
| outs: | |
| - path: artifacts/data_ingestion | |
| hash: md5 | |
| md5: 86510b1e2ff6da777357ccfdc278e4c8.dir | |
| size: 116493584 | |
| nfiles: 466 | |
| prepare_base_model: | |
| cmd: python src/cnnClassifier/pipeline/stage_02_prepare_base_model.py | |
| deps: | |
| - path: config/config.yaml | |
| hash: md5 | |
| md5: 20cd3ab789ce919b3687442bc4f2ab85 | |
| size: 1016 | |
| - path: params.yaml | |
| hash: md5 | |
| md5: 156a3b540bf80876a34e08b09faaf4fb | |
| size: 151 | |
| - path: src/cnnClassifier/components/prepare_base_model.py | |
| hash: md5 | |
| md5: 3c5230f332299193cb460420f3ce5057 | |
| size: 2063 | |
| - path: src/cnnClassifier/config/configuration.py | |
| hash: md5 | |
| md5: d41b12b3ad8d16ad963a9a34118da1ec | |
| size: 2873 | |
| - path: src/cnnClassifier/pipeline/stage_02_prepare_base_model.py | |
| hash: md5 | |
| md5: c886276ae57285dac8969b02bf9077ed | |
| size: 954 | |
| 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: d761158dc61a51df0233a4d98a02499f.dir | |
| size: 118073624 | |
| nfiles: 2 | |
| training: | |
| cmd: python src/cnnClassifier/pipeline/stage_03_model_trainer.py | |
| deps: | |
| - path: artifacts/data_ingestion/kidney-ct-scan-image | |
| hash: md5 | |
| md5: 33ed59dbe5dec8ce2bb8e489b55203e4.dir | |
| size: 58936381 | |
| nfiles: 465 | |
| - path: artifacts/prepare_base_model/base_model_updated.h5 | |
| hash: md5 | |
| md5: 12a1e3ebb90d89346ff2beb4fa21053b | |
| size: 59147544 | |
| - path: config/config.yaml | |
| hash: md5 | |
| md5: 20cd3ab789ce919b3687442bc4f2ab85 | |
| size: 1016 | |
| - path: params.yaml | |
| hash: md5 | |
| md5: 156a3b540bf80876a34e08b09faaf4fb | |
| size: 151 | |
| - path: src/cnnClassifier/components/model_trainer.py | |
| hash: md5 | |
| md5: bc19f92e2812f36ba12a7c66730a6e21 | |
| size: 2675 | |
| - path: src/cnnClassifier/config/configuration.py | |
| hash: md5 | |
| md5: d41b12b3ad8d16ad963a9a34118da1ec | |
| size: 2873 | |
| - path: src/cnnClassifier/pipeline/stage_03_model_trainer.py | |
| hash: md5 | |
| md5: 0951b497a475aac360e1c73b347b2295 | |
| size: 885 | |
| params: | |
| params.yaml: | |
| AUGMENTATION: true | |
| BATCH_SIZE: 16 | |
| EPOCHS: 5 | |
| IMAGE_SIZE: | |
| - 224 | |
| - 224 | |
| - 3 | |
| LEARNING_RATE: 0.01 | |
| outs: | |
| - path: artifacts/training/model.h5 | |
| hash: md5 | |
| md5: 87e2a46b9573a6bba1da41192f0dff18 | |
| size: 59147544 | |
| evaluation: | |
| cmd: python src/cnnClassifier/pipeline/stage_04_model_evaluation.py | |
| deps: | |
| - path: artifacts/training/model.h5 | |
| hash: md5 | |
| md5: 87e2a46b9573a6bba1da41192f0dff18 | |
| size: 59147544 | |
| - path: config/config.yaml | |
| hash: md5 | |
| md5: 20cd3ab789ce919b3687442bc4f2ab85 | |
| size: 1016 | |
| - path: src/cnnClassifier/components/model_evaluation_mlflow.py | |
| hash: md5 | |
| md5: 4612a6a44af8961549348656ece0c848 | |
| size: 2306 | |
| - path: src/cnnClassifier/config/configuration.py | |
| hash: md5 | |
| md5: d41b12b3ad8d16ad963a9a34118da1ec | |
| size: 2873 | |
| - path: src/cnnClassifier/pipeline/stage_04_model_evaluation.py | |
| hash: md5 | |
| md5: dff6dc7d6115804b764c98c53f3bbc43 | |
| size: 908 | |
| params: | |
| params.yaml: | |
| BATCH_SIZE: 16 | |
| IMAGE_SIZE: | |
| - 224 | |
| - 224 | |
| - 3 | |
| outs: | |
| - path: scores.json | |
| hash: md5 | |
| md5: 1fad9e68e2b1611a3fc59b064c62106e | |
| size: 72 | |