|
--- |
|
license: apache-2.0 |
|
metrics: |
|
- accuracy |
|
- f1 |
|
--- |
|
Return Marvel hero based on image. |
|
|
|
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 |
|
``` |