Spaces:
Sleeping
Sleeping
File size: 5,604 Bytes
f8f5549 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 |
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
|