Smolry commited on
Commit
92f4032
·
verified ·
1 Parent(s): ac41245

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +15 -6
README.md CHANGED
@@ -5,13 +5,22 @@ datasets:
5
  language:
6
  - en
7
  metrics:
8
- - f1
9
- - precision
10
- - recall
11
- - accuracy
 
 
 
 
 
 
 
 
12
  base_model:
13
  - google/efficientnet-b0
14
 
 
15
  pipeline_tag: image-classification
16
  tags:
17
  - computer-vision
@@ -120,7 +129,7 @@ Input Size:
120
 
121
  Dataset Repository:
122
 
123
- <HF_DATASET_LINK>
124
 
125
  Classes:
126
 
@@ -229,7 +238,7 @@ The model was trained in two stages:
229
  | ![Phase 1](training/phase1_loss.png) | ![Phase 2](training/phase2_loss.png) |
230
 
231
  The epoch-level training history is available in
232
- `training/training_history.json`.
233
 
234
  ---
235
 
 
5
  language:
6
  - en
7
  metrics:
8
+ - name: F1
9
+ type: f1
10
+ value: 0.86
11
+ - name: Precision
12
+ type: precision
13
+ value: 0.87
14
+ - name: Recall
15
+ type: recall
16
+ value: 0.85
17
+ - name: Accuracy
18
+ type: accuracy
19
+ value: 0.87
20
  base_model:
21
  - google/efficientnet-b0
22
 
23
+
24
  pipeline_tag: image-classification
25
  tags:
26
  - computer-vision
 
129
 
130
  Dataset Repository:
131
 
132
+ [![HSRP-Classification Dataset](https://img.shields.io/badge/Dataset-00000)](https://huggingface.co/datasets/Smolry/HSRP_classification_data)
133
 
134
  Classes:
135
 
 
238
  | ![Phase 1](training/phase1_loss.png) | ![Phase 2](training/phase2_loss.png) |
239
 
240
  The epoch-level training history is available in
241
+ [![Training Hisitory](https://img.shields.io/badge/Training_History-training__history.json-blue?logo=json&logoColor=white)](https://huggingface.co/Smolry/HSRP-classification/blob/main/training/training_history.json)
242
 
243
  ---
244