dima806's picture
Update README.md
43a549f 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.9614    0.9876    0.9743       403
       Angry     0.7440    0.6219    0.6775       402
       Happy     0.9100    0.9280    0.9189       403
     Neutral     0.5814    0.7618    0.6595       403
         Sad     0.6350    0.4318    0.5140       403
    Surprise     0.7577    0.8536    0.8028       403

    accuracy                         0.7642      2417
   macro avg     0.7649    0.7641    0.7578      2417
weighted avg     0.7649    0.7642    0.7579      2417