Hugging Face's logo

ThuyNT03
/
PhoBERT-cls-OCR

Text Classification
Transformers
PyTorch
roberta
Generated from Trainer
Inference Endpoints
Model card Files Community
2
ThuyNT03 commited on
Commit
df949d3
·
1 Parent(s): 62b7035

Training completed!

Browse files
Files changed (4) hide show
  1. README.md +12 -12
  2. config.json +1 -1
  3. pytorch_model.bin +2 -2
  4. training_args.bin +1 -1
README.md CHANGED
@@ -17,9 +17,9 @@ should probably proofread and complete it, then remove this comment. -->
17
 
18
  This model is a fine-tuned version of [vinai/phobert-base-v2](https://huggingface.co/vinai/phobert-base-v2) on the None dataset.
19
  It achieves the following results on the evaluation set:
20
- - Loss: 0.1546
21
- - Accuracy: 0.9593
22
- - F1: 0.9592
23
 
24
  ## Model description
25
 
@@ -50,19 +50,19 @@ The following hyperparameters were used during training:
50
 
51
  | Training Loss | Epoch | Step | Validation Loss | Accuracy | F1 |
52
  |:-------------:|:-----:|:----:|:---------------:|:--------:|:------:|
53
- | 0.4221 | 1.0 | 43 | 0.2568 | 0.9070 | 0.9049 |
54
- | 0.1993 | 2.0 | 86 | 0.1515 | 0.9593 | 0.9592 |
55
- | 0.1313 | 3.0 | 129 | 0.1582 | 0.9593 | 0.9591 |
56
- | 0.0966 | 4.0 | 172 | 0.1456 | 0.9651 | 0.9651 |
57
- | 0.0737 | 5.0 | 215 | 0.1432 | 0.9651 | 0.9651 |
58
- | 0.0592 | 6.0 | 258 | 0.1488 | 0.9651 | 0.9651 |
59
- | 0.0633 | 7.0 | 301 | 0.1605 | 0.9593 | 0.9592 |
60
- | 0.0575 | 8.0 | 344 | 0.1546 | 0.9593 | 0.9592 |
61
 
62
 
63
  ### Framework versions
64
 
65
- - Transformers 4.33.1
66
  - Pytorch 2.0.1+cu118
67
  - Datasets 2.14.5
68
  - Tokenizers 0.13.3
 
17
 
18
  This model is a fine-tuned version of [vinai/phobert-base-v2](https://huggingface.co/vinai/phobert-base-v2) on the None dataset.
19
  It achieves the following results on the evaluation set:
20
+ - Loss: 0.5182
21
+ - Accuracy: 0.8713
22
+ - F1: 0.8710
23
 
24
  ## Model description
25
 
 
50
 
51
  | Training Loss | Epoch | Step | Validation Loss | Accuracy | F1 |
52
  |:-------------:|:-----:|:----:|:---------------:|:--------:|:------:|
53
+ | 0.5631 | 1.0 | 25 | 0.4327 | 0.8119 | 0.8105 |
54
+ | 0.3123 | 2.0 | 50 | 0.4440 | 0.8020 | 0.7971 |
55
+ | 0.1806 | 3.0 | 75 | 0.4117 | 0.8515 | 0.8518 |
56
+ | 0.1107 | 4.0 | 100 | 0.4446 | 0.8614 | 0.8607 |
57
+ | 0.0729 | 5.0 | 125 | 0.4965 | 0.8713 | 0.8710 |
58
+ | 0.0473 | 6.0 | 150 | 0.4914 | 0.8812 | 0.8799 |
59
+ | 0.049 | 7.0 | 175 | 0.5021 | 0.8713 | 0.8710 |
60
+ | 0.0255 | 8.0 | 200 | 0.5182 | 0.8713 | 0.8710 |
61
 
62
 
63
  ### Framework versions
64
 
65
+ - Transformers 4.33.3
66
  - Pytorch 2.0.1+cu118
67
  - Datasets 2.14.5
68
  - Tokenizers 0.13.3
config.json CHANGED
@@ -22,7 +22,7 @@
22
  "problem_type": "single_label_classification",
23
  "tokenizer_class": "PhobertTokenizer",
24
  "torch_dtype": "float32",
25
- "transformers_version": "4.33.1",
26
  "type_vocab_size": 1,
27
  "use_cache": true,
28
  "vocab_size": 64001
 
22
  "problem_type": "single_label_classification",
23
  "tokenizer_class": "PhobertTokenizer",
24
  "torch_dtype": "float32",
25
+ "transformers_version": "4.33.3",
26
  "type_vocab_size": 1,
27
  "use_cache": true,
28
  "vocab_size": 64001
pytorch_model.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:1d674575aed0be876458616726f37b7691b9b384a6f208175e171e1c7c1af4c7
3
- size 540068209
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:919fabe275d0f9faac7ab49bb803adfd370fcb7cec818b89f3d119b29dfcc2ca
3
+ size 540065393
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:2e31ad726048ceeb69d241ff7f3d11d880e4aeb16d54079cb5c3454a4ed4cf8d
3
  size 4027
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6b502914df107074f2b1fee99e2fa70a06d7196e6eab915725fe0c427764d6df
3
  size 4027