evangeliazve commited on
Commit
0cffc77
1 Parent(s): a9318c2

update model card README.md

Browse files
Files changed (1) hide show
  1. README.md +16 -1
README.md CHANGED
@@ -2,6 +2,8 @@
2
  license: apache-2.0
3
  tags:
4
  - generated_from_trainer
 
 
5
  model-index:
6
  - name: predict_department_v2
7
  results: []
@@ -13,6 +15,9 @@ should probably proofread and complete it, then remove this comment. -->
13
  # predict_department_v2
14
 
15
  This model is a fine-tuned version of [bert-base-multilingual-uncased](https://huggingface.co/bert-base-multilingual-uncased) on the None dataset.
 
 
 
16
 
17
  ## Model description
18
 
@@ -37,7 +42,17 @@ The following hyperparameters were used during training:
37
  - seed: 42
38
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
39
  - lr_scheduler_type: linear
40
- - num_epochs: 2
 
 
 
 
 
 
 
 
 
 
41
 
42
  ### Framework versions
43
 
 
2
  license: apache-2.0
3
  tags:
4
  - generated_from_trainer
5
+ metrics:
6
+ - f1
7
  model-index:
8
  - name: predict_department_v2
9
  results: []
 
15
  # predict_department_v2
16
 
17
  This model is a fine-tuned version of [bert-base-multilingual-uncased](https://huggingface.co/bert-base-multilingual-uncased) on the None dataset.
18
+ It achieves the following results on the evaluation set:
19
+ - Loss: 0.3698
20
+ - F1: 0.9045
21
 
22
  ## Model description
23
 
 
42
  - seed: 42
43
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
44
  - lr_scheduler_type: linear
45
+ - num_epochs: 4
46
+
47
+ ### Training results
48
+
49
+ | Training Loss | Epoch | Step | Validation Loss | F1 |
50
+ |:-------------:|:-----:|:-----:|:---------------:|:------:|
51
+ | 0.3916 | 1.0 | 11724 | 0.3966 | 0.8752 |
52
+ | 0.3037 | 2.0 | 23448 | 0.3324 | 0.8948 |
53
+ | 0.2323 | 3.0 | 35172 | 0.3351 | 0.9023 |
54
+ | 0.1675 | 4.0 | 46896 | 0.3698 | 0.9045 |
55
+
56
 
57
  ### Framework versions
58