|
--- |
|
license: apache-2.0 |
|
metrics: |
|
- accuracy |
|
- f1 |
|
--- |
|
Predicts face emotion based on facial image. |
|
|
|
See https://www.kaggle.com/code/dima806/face-emotions-image-detection-vit for more details. |
|
|
|
``` |
|
Classification report: |
|
|
|
precision recall f1-score support |
|
|
|
Ahegao 0.9738 0.9919 0.9828 1611 |
|
Angry 0.8439 0.6580 0.7394 1611 |
|
Happy 0.8939 0.9261 0.9098 1611 |
|
Neutral 0.6056 0.7635 0.6755 1611 |
|
Sad 0.6661 0.5140 0.5802 1611 |
|
Surprise 0.7704 0.8733 0.8186 1610 |
|
|
|
accuracy 0.7878 9665 |
|
macro avg 0.7923 0.7878 0.7844 9665 |
|
weighted avg 0.7923 0.7878 0.7844 9665 |
|
``` |