dima806's picture
Update README.md
829b17a verified
|
raw
history blame
765 Bytes
metadata
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.9767    0.9876    0.9821      1611
       Angry     0.8219    0.6019    0.6949      1610
       Happy     0.8939    0.9255    0.9094      1611
     Neutral     0.6097    0.7778    0.6836      1611
         Sad     0.6404    0.5053    0.5649      1611
    Surprise     0.7595    0.8783    0.8146      1611

    accuracy                         0.7794      9665
   macro avg     0.7837    0.7794    0.7749      9665
weighted avg     0.7837    0.7794    0.7749      9665