H-Layba commited on
Commit
bd98698
·
verified ·
1 Parent(s): 6c62a78

Fine-tuned mBERT for Urdu sentiment classification — corrected labels

Browse files
Files changed (1) hide show
  1. README.md +8 -8
README.md CHANGED
@@ -19,9 +19,9 @@ should probably proofread and complete it, then remove this comment. -->
19
 
20
  This model is a fine-tuned version of [bert-base-multilingual-cased](https://huggingface.co/bert-base-multilingual-cased) on an unknown dataset.
21
  It achieves the following results on the evaluation set:
22
- - Loss: 0.8915
23
- - Accuracy: 0.81
24
- - F1: 0.8098
25
 
26
  ## Model description
27
 
@@ -54,11 +54,11 @@ The following hyperparameters were used during training:
54
 
55
  | Training Loss | Epoch | Step | Validation Loss | Accuracy | F1 |
56
  |:-------------:|:-----:|:----:|:---------------:|:--------:|:------:|
57
- | 0.8954 | 1.0 | 625 | 0.9527 | 0.7696 | 0.7656 |
58
- | 0.7918 | 2.0 | 1250 | 0.8385 | 0.8 | 0.7999 |
59
- | 0.6211 | 3.0 | 1875 | 0.8908 | 0.8101 | 0.8099 |
60
- | 0.5833 | 4.0 | 2500 | 0.9350 | 0.8088 | 0.8087 |
61
- | 0.4515 | 5.0 | 3125 | 1.0381 | 0.8081 | 0.8081 |
62
 
63
 
64
  ### Framework versions
 
19
 
20
  This model is a fine-tuned version of [bert-base-multilingual-cased](https://huggingface.co/bert-base-multilingual-cased) on an unknown dataset.
21
  It achieves the following results on the evaluation set:
22
+ - Loss: 1.0358
23
+ - Accuracy: 0.8074
24
+ - F1: 0.8073
25
 
26
  ## Model description
27
 
 
54
 
55
  | Training Loss | Epoch | Step | Validation Loss | Accuracy | F1 |
56
  |:-------------:|:-----:|:----:|:---------------:|:--------:|:------:|
57
+ | 0.8904 | 1.0 | 625 | 0.9830 | 0.7664 | 0.7622 |
58
+ | 0.7935 | 2.0 | 1250 | 0.8413 | 0.7978 | 0.7975 |
59
+ | 0.6253 | 3.0 | 1875 | 0.9115 | 0.8021 | 0.8014 |
60
+ | 0.5877 | 4.0 | 2500 | 0.9304 | 0.8055 | 0.8051 |
61
+ | 0.4309 | 5.0 | 3125 | 1.0358 | 0.8074 | 0.8073 |
62
 
63
 
64
  ### Framework versions