Mitchins commited on
Commit
e71505a
·
verified ·
1 Parent(s): 031e4c2

Upload folder using huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +15 -15
README.md CHANGED
@@ -1,6 +1,10 @@
1
  ---
2
  language: en
3
  license: mit
 
 
 
 
4
  tags:
5
  - image-classification
6
  - anime
@@ -9,45 +13,41 @@ tags:
9
  - pytorch
10
  - safetensors
11
  - multi-class
 
 
12
  framework: pytorch
13
  library_name: torchvision
14
  pipeline_tag: image-classification
 
15
  model-index:
16
  - name: anime-style-classifier-efficientnet-b0
17
- base_model:
18
- - google/efficientnet-b0
19
  results:
20
  - task:
21
  type: image-classification
22
  dataset:
23
  name: synthetic-anime-scenes
24
- type: synthetic
 
25
  metrics:
26
- - name: Accuracy (validation)
27
  type: accuracy
28
  value: 0.9818
29
- - name: Macro F1 (validation)
30
  type: f1
31
  value: 0.9879
32
- source:
33
- name: Repository
34
- url: https://github.com/Mitchins/anime-style-classifier-efficientnet-b0
35
  - task:
36
  type: image-classification
37
  dataset:
38
  name: synthetic-anime-scenes
39
- type: synthetic
 
40
  metrics:
41
- - name: Accuracy (holdout)
42
  type: accuracy
43
  value: 1.0
44
- - name: Macro F1 (holdout)
45
  type: f1
46
  value: 1.0
47
- source:
48
- name: Hugging Face model page
49
- url: https://huggingface.co/Mitchins/anime-style-classifier-efficientnet-b0
50
- usage: "See inference.py"
51
  ---
52
 
53
  # Anime Style Classifier - EfficientNet-B0
 
1
  ---
2
  language: en
3
  license: mit
4
+ base_model: google/efficientnet-b0 # <-- top-level so it shows in the sidebar
5
+ datasets:
6
+ - synthetic-anime-scenes # optional; helps sidebar/search
7
+
8
  tags:
9
  - image-classification
10
  - anime
 
13
  - pytorch
14
  - safetensors
15
  - multi-class
16
+ - "base_model:google/efficientnet-b0" # optional searchable tag
17
+
18
  framework: pytorch
19
  library_name: torchvision
20
  pipeline_tag: image-classification
21
+
22
  model-index:
23
  - name: anime-style-classifier-efficientnet-b0
 
 
24
  results:
25
  - task:
26
  type: image-classification
27
  dataset:
28
  name: synthetic-anime-scenes
29
+ type: synthetic # <-- REQUIRED
30
+ split: validation
31
  metrics:
32
+ - name: Accuracy
33
  type: accuracy
34
  value: 0.9818
35
+ - name: Macro F1
36
  type: f1
37
  value: 0.9879
 
 
 
38
  - task:
39
  type: image-classification
40
  dataset:
41
  name: synthetic-anime-scenes
42
+ type: synthetic # <-- REQUIRED
43
+ split: holdout
44
  metrics:
45
+ - name: Accuracy
46
  type: accuracy
47
  value: 1.0
48
+ - name: Macro F1
49
  type: f1
50
  value: 1.0
 
 
 
 
51
  ---
52
 
53
  # Anime Style Classifier - EfficientNet-B0