| | --- |
| | 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 |
| | ``` |
