bergurth commited on
Commit
d4a7bf0
1 Parent(s): 6fb857c

update model card README.md

Browse files
Files changed (1) hide show
  1. README.md +41 -1
README.md CHANGED
@@ -4,9 +4,34 @@ tags:
4
  - generated_from_trainer
5
  datasets:
6
  - mim_gold_ner
 
 
 
 
 
7
  model-index:
8
  - name: IceBERT-finetuned-ner
9
- results: []
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
10
  ---
11
 
12
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -15,6 +40,12 @@ should probably proofread and complete it, then remove this comment. -->
15
  # IceBERT-finetuned-ner
16
 
17
  This model is a fine-tuned version of [vesteinn/IceBERT](https://huggingface.co/vesteinn/IceBERT) on the mim_gold_ner dataset.
 
 
 
 
 
 
18
 
19
  ## Model description
20
 
@@ -41,6 +72,15 @@ The following hyperparameters were used during training:
41
  - lr_scheduler_type: linear
42
  - num_epochs: 3
43
 
 
 
 
 
 
 
 
 
 
44
  ### Framework versions
45
 
46
  - Transformers 4.11.2
 
4
  - generated_from_trainer
5
  datasets:
6
  - mim_gold_ner
7
+ metrics:
8
+ - precision
9
+ - recall
10
+ - f1
11
+ - accuracy
12
  model-index:
13
  - name: IceBERT-finetuned-ner
14
+ results:
15
+ - task:
16
+ name: Token Classification
17
+ type: token-classification
18
+ dataset:
19
+ name: mim_gold_ner
20
+ type: mim_gold_ner
21
+ args: mim-gold-ner
22
+ metrics:
23
+ - name: Precision
24
+ type: precision
25
+ value: 0.8873049035270985
26
+ - name: Recall
27
+ type: recall
28
+ value: 0.8627076114231091
29
+ - name: F1
30
+ type: f1
31
+ value: 0.8748333939173634
32
+ - name: Accuracy
33
+ type: accuracy
34
+ value: 0.9848076353832492
35
  ---
36
 
37
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
 
40
  # IceBERT-finetuned-ner
41
 
42
  This model is a fine-tuned version of [vesteinn/IceBERT](https://huggingface.co/vesteinn/IceBERT) on the mim_gold_ner dataset.
43
+ It achieves the following results on the evaluation set:
44
+ - Loss: 0.0783
45
+ - Precision: 0.8873
46
+ - Recall: 0.8627
47
+ - F1: 0.8748
48
+ - Accuracy: 0.9848
49
 
50
  ## Model description
51
 
 
72
  - lr_scheduler_type: linear
73
  - num_epochs: 3
74
 
75
+ ### Training results
76
+
77
+ | Training Loss | Epoch | Step | Validation Loss | Precision | Recall | F1 | Accuracy |
78
+ |:-------------:|:-----:|:----:|:---------------:|:---------:|:------:|:------:|:--------:|
79
+ | 0.0539 | 1.0 | 2904 | 0.0768 | 0.8732 | 0.8453 | 0.8590 | 0.9833 |
80
+ | 0.0281 | 2.0 | 5808 | 0.0737 | 0.8781 | 0.8492 | 0.8634 | 0.9838 |
81
+ | 0.0166 | 3.0 | 8712 | 0.0783 | 0.8873 | 0.8627 | 0.8748 | 0.9848 |
82
+
83
+
84
  ### Framework versions
85
 
86
  - Transformers 4.11.2