dima806's picture
Update README.md
0fb9cb3 verified
metadata
license: apache-2.0
metrics:
  - accuracy
base_model:
  - google/vit-base-patch16-224-in21k

Returns person celebrity-style attractiveness (0 to 1) based on facial image with about 83% accuracy.

See https://www.kaggle.com/code/dima806/attractive-faces-celebs-detection-vit for more details.

image/png

Classification report:

                precision    recall  f1-score   support

    attractive     0.8297    0.8502    0.8398      5192
not attractive     0.8464    0.8255    0.8358      5192

      accuracy                         0.8378     10384
     macro avg     0.8380    0.8378    0.8378     10384
  weighted avg     0.8380    0.8378    0.8378     10384