dima806 commited on
Commit
a8e129d
1 Parent(s): 72e0b81

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +7 -7
README.md CHANGED
@@ -9,21 +9,21 @@ base_model:
9
  - google/vit-base-patch16-224-in21k
10
  pipeline_tag: image-classification
11
  ---
12
- Detects gender with about 88% accuracy based on an image.
13
 
14
  See https://www.kaggle.com/code/dima806/gender-image-classification-vit for details.
15
 
16
- ![image/png](https://cdn-uploads.huggingface.co/production/uploads/6449300e3adf50d864095b90/NLKajM7gBl16afwtMGLpD.png)
17
 
18
  ```
19
  Classification report:
20
 
21
  precision recall f1-score support
22
 
23
- Female 0.8563 0.8938 0.8746 480
24
- Male 0.8978 0.8615 0.8793 520
25
 
26
- accuracy 0.8770 1000
27
- macro avg 0.8770 0.8776 0.8770 1000
28
- weighted avg 0.8779 0.8770 0.8770 1000
29
  ```
 
9
  - google/vit-base-patch16-224-in21k
10
  pipeline_tag: image-classification
11
  ---
12
+ Detects gender with about 93.4% accuracy based on an image.
13
 
14
  See https://www.kaggle.com/code/dima806/gender-image-classification-vit for details.
15
 
16
+ ![image/png](https://cdn-uploads.huggingface.co/production/uploads/6449300e3adf50d864095b90/OOerkHOqKZsJ4F4CZsZEW.png)
17
 
18
  ```
19
  Classification report:
20
 
21
  precision recall f1-score support
22
 
23
+ Female 0.9531 0.9051 0.9285 4720
24
+ Male 0.9188 0.9602 0.9391 5280
25
 
26
+ accuracy 0.9342 10000
27
+ macro avg 0.9360 0.9327 0.9338 10000
28
+ weighted avg 0.9350 0.9342 0.9341 10000
29
  ```