Hugging Face's logo

ThuyNT03
/
PhoBERT-cls-OCR

Text Classification
Transformers
PyTorch
roberta
Generated from Trainer
Inference Endpoints
Model card Files Community
1
ThuyNT03 commited on
Commit
27f4afc
•
1 Parent(s): 8d7c073

Training completed!

Browse files
Files changed (4) hide show
  1. README.md +13 -16
  2. config.json +2 -1
  3. pytorch_model.bin +1 -1
  4. training_args.bin +1 -1
README.md CHANGED
@@ -1,5 +1,5 @@
1
  ---
2
- base_model: vinai/phobert-base-v2
3
  tags:
4
  - generated_from_trainer
5
  metrics:
@@ -15,11 +15,11 @@ should probably proofread and complete it, then remove this comment. -->
15
 
16
  # PhoBERT-cls-OCR
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.6475
21
- - Accuracy: 0.8416
22
- - F1: 0.8422
23
 
24
  ## Model description
25
 
@@ -44,22 +44,19 @@ The following hyperparameters were used during training:
44
  - seed: 42
45
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
46
  - lr_scheduler_type: linear
47
- - num_epochs: 10
48
 
49
  ### Training results
50
 
51
  | Training Loss | Epoch | Step | Validation Loss | Accuracy | F1 |
52
  |:-------------:|:-----:|:----:|:---------------:|:--------:|:------:|
53
- | 0.5849 | 1.0 | 25 | 0.4385 | 0.8218 | 0.8225 |
54
- | 0.3382 | 2.0 | 50 | 0.3785 | 0.8614 | 0.8599 |
55
- | 0.1897 | 3.0 | 75 | 0.4339 | 0.8515 | 0.8511 |
56
- | 0.1101 | 4.0 | 100 | 0.4626 | 0.8614 | 0.8599 |
57
- | 0.0798 | 5.0 | 125 | 0.5006 | 0.8713 | 0.8703 |
58
- | 0.0547 | 6.0 | 150 | 0.5670 | 0.8614 | 0.8614 |
59
- | 0.034 | 7.0 | 175 | 0.5568 | 0.8812 | 0.8806 |
60
- | 0.0121 | 8.0 | 200 | 0.6414 | 0.8317 | 0.8326 |
61
- | 0.0207 | 9.0 | 225 | 0.6713 | 0.8416 | 0.8427 |
62
- | 0.0203 | 10.0 | 250 | 0.6475 | 0.8416 | 0.8422 |
63
 
64
 
65
  ### Framework versions
 
1
  ---
2
+ base_model: vinai/phobert-base
3
  tags:
4
  - generated_from_trainer
5
  metrics:
 
15
 
16
  # PhoBERT-cls-OCR
17
 
18
+ This model is a fine-tuned version of [vinai/phobert-base](https://huggingface.co/vinai/phobert-base) on the None dataset.
19
  It achieves the following results on the evaluation set:
20
+ - Loss: 0.4288
21
+ - Accuracy: 0.8812
22
+ - F1: 0.8812
23
 
24
  ## Model description
25
 
 
44
  - seed: 42
45
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
46
  - lr_scheduler_type: linear
47
+ - num_epochs: 7
48
 
49
  ### Training results
50
 
51
  | Training Loss | Epoch | Step | Validation Loss | Accuracy | F1 |
52
  |:-------------:|:-----:|:----:|:---------------:|:--------:|:------:|
53
+ | 0.5728 | 1.0 | 25 | 0.4112 | 0.8614 | 0.8590 |
54
+ | 0.2892 | 2.0 | 50 | 0.3444 | 0.8515 | 0.8511 |
55
+ | 0.1954 | 3.0 | 75 | 0.3638 | 0.8812 | 0.8816 |
56
+ | 0.1387 | 4.0 | 100 | 0.3591 | 0.8812 | 0.8806 |
57
+ | 0.1029 | 5.0 | 125 | 0.3809 | 0.8911 | 0.8908 |
58
+ | 0.053 | 6.0 | 150 | 0.4145 | 0.8911 | 0.8903 |
59
+ | 0.0527 | 7.0 | 175 | 0.4288 | 0.8812 | 0.8812 |
 
 
 
60
 
61
 
62
  ### Framework versions
config.json CHANGED
@@ -1,5 +1,5 @@
1
  {
2
- "_name_or_path": "vinai/phobert-base-v2",
3
  "architectures": [
4
  "RobertaForSequenceClassification"
5
  ],
@@ -7,6 +7,7 @@
7
  "bos_token_id": 0,
8
  "classifier_dropout": null,
9
  "eos_token_id": 2,
 
10
  "hidden_act": "gelu",
11
  "hidden_dropout_prob": 0.1,
12
  "hidden_size": 768,
 
1
  {
2
+ "_name_or_path": "vinai/phobert-base",
3
  "architectures": [
4
  "RobertaForSequenceClassification"
5
  ],
 
7
  "bos_token_id": 0,
8
  "classifier_dropout": null,
9
  "eos_token_id": 2,
10
+ "gradient_checkpointing": false,
11
  "hidden_act": "gelu",
12
  "hidden_dropout_prob": 0.1,
13
  "hidden_size": 768,
pytorch_model.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:a8511396f9ea5184235dd616f8f03bb00371169732c46fca287c6c7ce39b5ce1
3
  size 540068209
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f4d680c566094d14f0c7d2318c1469000748ad8e0e4770240d47159617aa60fb
3
  size 540068209
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:73494ef25fd4750f7db3c8f601fa2adece4cbce7b471d0e3d9a1dfecdb6bbf52
3
  size 4027
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:80326e334c5466457ee484bccbf56053310faea7cd1288387e41491a6158429b
3
  size 4027