Update README.md
Browse files
README.md
CHANGED
|
@@ -4,4 +4,25 @@ metrics:
|
|
| 4 |
- accuracy
|
| 5 |
- f1
|
| 6 |
---
|
| 7 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 4 |
- accuracy
|
| 5 |
- f1
|
| 6 |
---
|
| 7 |
+
Return Marvel hero based on image.
|
| 8 |
+
|
| 9 |
+
See https://www.kaggle.com/code/dima806/marvel-heroes-image-detection-vit for more details.
|
| 10 |
+
|
| 11 |
+
```
|
| 12 |
+
Classification report:
|
| 13 |
+
|
| 14 |
+
precision recall f1-score support
|
| 15 |
+
|
| 16 |
+
captain america 0.8519 0.8519 0.8519 162
|
| 17 |
+
black widow 0.8634 0.8528 0.8580 163
|
| 18 |
+
spider-man 0.9571 0.9630 0.9600 162
|
| 19 |
+
thanos 0.8917 0.8589 0.8750 163
|
| 20 |
+
ironman 0.8614 0.8827 0.8720 162
|
| 21 |
+
hulk 0.8889 0.8395 0.8635 162
|
| 22 |
+
loki 0.8957 0.8957 0.8957 163
|
| 23 |
+
doctor strange 0.8629 0.9264 0.8935 163
|
| 24 |
+
|
| 25 |
+
accuracy 0.8838 1300
|
| 26 |
+
macro avg 0.8841 0.8838 0.8837 1300
|
| 27 |
+
weighted avg 0.8841 0.8838 0.8837 1300
|
| 28 |
+
```
|