| { | |
| "model_type": "cnn_emotion_classifier", | |
| "num_classes": 9, | |
| "class_names": [ | |
| "angry", | |
| "disgust", | |
| "fear", | |
| "happy", | |
| "neutral", | |
| "sad", | |
| "surprise", | |
| "contempt", | |
| "class_8" | |
| ], | |
| "input_shape": [ | |
| 48, | |
| 48, | |
| 1 | |
| ], | |
| "dataset": "8-facial-expressions-for-yolo", | |
| "framework": "tensorflow", | |
| "training_config": { | |
| "optimizer": "adam", | |
| "loss": "categorical_crossentropy", | |
| "metrics": [ | |
| "accuracy" | |
| ], | |
| "epochs": 20, | |
| "batch_size": 32 | |
| } | |
| } |