metadata
license: apache-2.0
metrics:
- accuracy
- f1
base_model:
- google/vit-base-patch16-224-in21k
Return Marvel hero based on image with about 88% accuracy.
See https://www.kaggle.com/code/dima806/marvel-heroes-image-detection-vit for more details.
Classification report:
precision recall f1-score support
captain america 0.8519 0.8519 0.8519 162
black widow 0.8634 0.8528 0.8580 163
spider-man 0.9571 0.9630 0.9600 162
thanos 0.8917 0.8589 0.8750 163
ironman 0.8614 0.8827 0.8720 162
hulk 0.8889 0.8395 0.8635 162
loki 0.8957 0.8957 0.8957 163
doctor strange 0.8629 0.9264 0.8935 163
accuracy 0.8838 1300
macro avg 0.8841 0.8838 0.8837 1300
weighted avg 0.8841 0.8838 0.8837 1300