Spaces:
Running
Running
| schema: '2.0' | |
| stages: | |
| data_validation: | |
| cmd: python src/EmotionRecognition/pipeline/stage_02_data_validation.py | |
| deps: | |
| - path: artifacts/data_ingestion | |
| hash: md5 | |
| md5: 9208f64defb6697b78bab62e943d955d.dir | |
| size: 302675528 | |
| nfiles: 2 | |
| - path: src/EmotionRecognition/config/configuration.py | |
| hash: md5 | |
| md5: dacf4230e18681185b786aa280cdec5e | |
| size: 4275 | |
| - path: src/EmotionRecognition/pipeline/stage_02_data_validation.py | |
| hash: md5 | |
| md5: 18a3d78c83dc5b278e14523077035e41 | |
| size: 1141 | |
| outs: | |
| - path: artifacts/data_validation/status.txt | |
| hash: md5 | |
| md5: 86e6a2f694c57a675b3e2da6b95ff9ba | |
| size: 23 | |
| data_preparation: | |
| cmd: python src/EmotionRecognition/pipeline/stage_01_data_preparation.py | |
| deps: | |
| - path: data/raw/CK+48 | |
| hash: md5 | |
| md5: a1559eddfd0d86b541e5df18b4b8205e.dir | |
| size: 1715162 | |
| nfiles: 981 | |
| - path: data/raw/fer2013.csv | |
| hash: md5 | |
| md5: f8428a1edbd21e88f42c73edd2a14f95 | |
| size: 301072766 | |
| - path: data/raw/fer2013new.csv | |
| hash: md5 | |
| md5: 413eba86d6e454536b99705b8c7fc5c5 | |
| size: 1602762 | |
| - path: src/EmotionRecognition/components/data_preparation.py | |
| hash: md5 | |
| md5: 228140227aaedb9f07b4c00462f267c6 | |
| size: 5776 | |
| - path: src/EmotionRecognition/config/configuration.py | |
| hash: md5 | |
| md5: 8786c8d41e2e50a49b4ca6d5bf59ad44 | |
| size: 2910 | |
| - path: src/EmotionRecognition/pipeline/stage_01_data_preparation.py | |
| hash: md5 | |
| md5: 1a324b8f1cf01e4e60e0a8529b23b577 | |
| size: 1110 | |
| params: | |
| params.yaml: | |
| DATA_PARAMS.CLASSES: | |
| - angry | |
| - disgust | |
| - fear | |
| - happy | |
| - neutral | |
| - sad | |
| - surprise | |
| outs: | |
| - path: artifacts/data_preparation/test | |
| hash: md5 | |
| md5: 79c105a50ccbe2557fea9fab2c743fa5.dir | |
| size: 6249935 | |
| nfiles: 3589 | |
| - path: artifacts/data_preparation/train | |
| hash: md5 | |
| md5: 750c0a305d28467341396ab591ed2731.dir | |
| size: 51232879 | |
| nfiles: 29471 | |
| model_training: | |
| cmd: python src/EmotionRecognition/pipeline/stage_02_model_training.py | |
| deps: | |
| - path: artifacts/data_preparation/test | |
| hash: md5 | |
| md5: 79c105a50ccbe2557fea9fab2c743fa5.dir | |
| size: 6249935 | |
| nfiles: 3589 | |
| - path: artifacts/data_preparation/train | |
| hash: md5 | |
| md5: 750c0a305d28467341396ab591ed2731.dir | |
| size: 51232879 | |
| nfiles: 29471 | |
| - path: src/EmotionRecognition/components/model_trainer.py | |
| hash: md5 | |
| md5: 5192acef195c9a9b03a88490476ead1c | |
| size: 3916 | |
| - path: src/EmotionRecognition/pipeline/stage_02_model_training.py | |
| hash: md5 | |
| md5: 2ee36d6e30a3a262e8327a26e71a37e9 | |
| size: 1076 | |
| params: | |
| params.yaml: | |
| DATA_PARAMS: | |
| IMAGE_SIZE: | |
| - 224 | |
| - 224 | |
| CHANNELS: 3 | |
| BATCH_SIZE: 32 | |
| CLASSES: | |
| - angry | |
| - disgust | |
| - fear | |
| - happy | |
| - neutral | |
| - sad | |
| - surprise | |
| NUM_CLASSES: 7 | |
| TRAINING_PARAMS: | |
| EPOCHS: 50 | |
| LEARNING_RATE: 0.0001 | |
| OPTIMIZER: Adam | |
| LOSS_FUNCTION: CategoricalCrossentropy | |
| METRICS: | |
| - accuracy | |
| DROPOUT_RATE: 0.5 | |
| outs: | |
| - path: artifacts/training/model.keras | |
| hash: md5 | |
| md5: 2c632cb4cbf3f2944145a8da1927f2cf | |
| size: 11331400 | |
| model_evaluation: | |
| cmd: python src/EmotionRecognition/pipeline/stage_03_model_evaluation.py | |
| deps: | |
| - path: artifacts/data_preparation/test | |
| hash: md5 | |
| md5: 79c105a50ccbe2557fea9fab2c743fa5.dir | |
| size: 6249935 | |
| nfiles: 3589 | |
| - path: artifacts/training/model.keras | |
| hash: md5 | |
| md5: 2c632cb4cbf3f2944145a8da1927f2cf | |
| size: 11331400 | |
| - path: src/EmotionRecognition/components/model_evaluation.py | |
| hash: md5 | |
| md5: 8b327667db406dd7c6489937747b8537 | |
| size: 2429 | |
| params: | |
| params.yaml: | |
| DATA_PARAMS: | |
| IMAGE_SIZE: | |
| - 224 | |
| - 224 | |
| CHANNELS: 3 | |
| BATCH_SIZE: 32 | |
| CLASSES: | |
| - angry | |
| - disgust | |
| - fear | |
| - happy | |
| - neutral | |
| - sad | |
| - surprise | |
| NUM_CLASSES: 7 | |
| outs: | |
| - path: artifacts/evaluation/metrics.json | |
| hash: md5 | |
| md5: 3e8f938b34095f56c597110c5d86064e | |
| size: 72 | |
| data_preprocessing: | |
| cmd: python src/EmotionRecognition/pipeline/stage_02_data_preprocessing.py | |
| deps: | |
| - path: artifacts/data_preparation/test | |
| hash: md5 | |
| md5: 79c105a50ccbe2557fea9fab2c743fa5.dir | |
| size: 6249935 | |
| nfiles: 3589 | |
| - path: artifacts/data_preparation/train | |
| hash: md5 | |
| md5: 750c0a305d28467341396ab591ed2731.dir | |
| size: 51232879 | |
| nfiles: 29471 | |
| - path: src/EmotionRecognition/components/data_preprocessing.py | |
| hash: md5 | |
| md5: bc85964fdf86afb289051c2498037eb8 | |
| size: 3903 | |
| - path: src/EmotionRecognition/pipeline/stage_02_data_preprocessing.py | |
| hash: md5 | |
| md5: 5631296a6b7bace5c2f6979eda5ca081 | |
| size: 971 | |
| params: | |
| params.yaml: | |
| DATA_PARAMS.CLASSES: | |
| - angry | |
| - disgust | |
| - fear | |
| - happy | |
| - neutral | |
| - sad | |
| - surprise | |
| outs: | |
| - path: artifacts/data_preprocessing/test | |
| hash: md5 | |
| md5: 79c105a50ccbe2557fea9fab2c743fa5.dir | |
| size: 6249935 | |
| nfiles: 3589 | |
| - path: artifacts/data_preprocessing/train | |
| hash: md5 | |
| md5: 3dc8382a4774d1a1f1d1e5dfe3ca4c1b.dir | |
| size: 18389122 | |
| nfiles: 10500 | |