dima806 commited on
Commit
e2e71a2
1 Parent(s): 0f422da

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +22 -1
README.md CHANGED
@@ -4,4 +4,25 @@ metrics:
4
  - accuracy
5
  - f1
6
  ---
7
- See https://www.kaggle.com/code/dima806/marvel-heroes-image-detection-vit for more details.
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
+ ```