trng1305 commited on
Commit
c0456c3
·
1 Parent(s): 4a1fc14

End of training

Browse files
README.md ADDED
@@ -0,0 +1,79 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ tags:
3
+ - generated_from_trainer
4
+ datasets:
5
+ - sroie
6
+ model-index:
7
+ - name: layoutlm-sroie
8
+ results: []
9
+ ---
10
+
11
+ <!-- This model card has been generated automatically according to the information the Trainer had access to. You
12
+ should probably proofread and complete it, then remove this comment. -->
13
+
14
+ # layoutlm-sroie
15
+
16
+ This model is a fine-tuned version of [microsoft/layoutlm-base-uncased](https://huggingface.co/microsoft/layoutlm-base-uncased) on the sroie dataset.
17
+ It achieves the following results on the evaluation set:
18
+ - Loss: 0.0320
19
+ - Address: {'precision': 0.9154929577464789, 'recall': 0.9365994236311239, 'f1': 0.925925925925926, 'number': 347}
20
+ - Company: {'precision': 0.9228650137741047, 'recall': 0.9654178674351584, 'f1': 0.9436619718309859, 'number': 347}
21
+ - Date: {'precision': 0.9884393063583815, 'recall': 0.9855907780979827, 'f1': 0.987012987012987, 'number': 347}
22
+ - Total: {'precision': 0.8431372549019608, 'recall': 0.8674351585014409, 'f1': 0.8551136363636365, 'number': 347}
23
+ - Overall Precision: 0.9170
24
+ - Overall Recall: 0.9388
25
+ - Overall F1: 0.9277
26
+ - Overall Accuracy: 0.9942
27
+
28
+ ## Model description
29
+
30
+ More information needed
31
+
32
+ ## Intended uses & limitations
33
+
34
+ More information needed
35
+
36
+ ## Training and evaluation data
37
+
38
+ More information needed
39
+
40
+ ## Training procedure
41
+
42
+ ### Training hyperparameters
43
+
44
+ The following hyperparameters were used during training:
45
+ - learning_rate: 3e-05
46
+ - train_batch_size: 16
47
+ - eval_batch_size: 8
48
+ - seed: 42
49
+ - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
50
+ - lr_scheduler_type: linear
51
+ - num_epochs: 15
52
+
53
+ ### Training results
54
+
55
+ | Training Loss | Epoch | Step | Validation Loss | Address | Company | Date | Total | Overall Precision | Overall Recall | Overall F1 | Overall Accuracy |
56
+ |:-------------:|:-----:|:----:|:---------------:|:--------------------------------------------------------------------------------------------------------:|:--------------------------------------------------------------------------------------------------------:|:--------------------------------------------------------------------------------------------------------:|:----------------------------------------------------------------------------------------------------------:|:-----------------:|:--------------:|:----------:|:----------------:|
57
+ | 0.5392 | 1.0 | 40 | 0.1200 | {'precision': 0.8346883468834688, 'recall': 0.8876080691642652, 'f1': 0.8603351955307261, 'number': 347} | {'precision': 0.75, 'recall': 0.8126801152737753, 'f1': 0.7800829875518672, 'number': 347} | {'precision': 0.648068669527897, 'recall': 0.8703170028818443, 'f1': 0.7429274292742927, 'number': 347} | {'precision': 0.0, 'recall': 0.0, 'f1': 0.0, 'number': 347} | 0.7366 | 0.6427 | 0.6864 | 0.9712 |
58
+ | 0.0803 | 2.0 | 80 | 0.0485 | {'precision': 0.896358543417367, 'recall': 0.9221902017291066, 'f1': 0.9090909090909091, 'number': 347} | {'precision': 0.8897849462365591, 'recall': 0.9538904899135446, 'f1': 0.9207232267037552, 'number': 347} | {'precision': 0.9628571428571429, 'recall': 0.9711815561959655, 'f1': 0.9670014347202295, 'number': 347} | {'precision': 0.5219298245614035, 'recall': 0.34293948126801155, 'f1': 0.41391304347826086, 'number': 347} | 0.8470 | 0.7976 | 0.8215 | 0.9876 |
59
+ | 0.0441 | 3.0 | 120 | 0.0373 | {'precision': 0.9131652661064426, 'recall': 0.9394812680115274, 'f1': 0.9261363636363636, 'number': 347} | {'precision': 0.9277777777777778, 'recall': 0.962536023054755, 'f1': 0.9448373408769449, 'number': 347} | {'precision': 0.9826589595375722, 'recall': 0.9798270893371758, 'f1': 0.9812409812409811, 'number': 347} | {'precision': 0.5418502202643172, 'recall': 0.7089337175792507, 'f1': 0.6142322097378278, 'number': 347} | 0.8214 | 0.8977 | 0.8578 | 0.9897 |
60
+ | 0.0309 | 4.0 | 160 | 0.0333 | {'precision': 0.8864265927977839, 'recall': 0.9221902017291066, 'f1': 0.903954802259887, 'number': 347} | {'precision': 0.915068493150685, 'recall': 0.962536023054755, 'f1': 0.9382022471910113, 'number': 347} | {'precision': 0.9912790697674418, 'recall': 0.9827089337175793, 'f1': 0.9869753979739508, 'number': 347} | {'precision': 0.6556122448979592, 'recall': 0.7406340057636888, 'f1': 0.6955345060893099, 'number': 347} | 0.8564 | 0.9020 | 0.8786 | 0.9916 |
61
+ | 0.0249 | 5.0 | 200 | 0.0318 | {'precision': 0.907563025210084, 'recall': 0.9337175792507204, 'f1': 0.9204545454545455, 'number': 347} | {'precision': 0.9054054054054054, 'recall': 0.9654178674351584, 'f1': 0.9344490934449093, 'number': 347} | {'precision': 0.9912536443148688, 'recall': 0.9798270893371758, 'f1': 0.9855072463768115, 'number': 347} | {'precision': 0.7186700767263428, 'recall': 0.8097982708933718, 'f1': 0.7615176151761518, 'number': 347} | 0.8761 | 0.9222 | 0.8986 | 0.9924 |
62
+ | 0.0183 | 6.0 | 240 | 0.0294 | {'precision': 0.9103641456582633, 'recall': 0.9365994236311239, 'f1': 0.9232954545454545, 'number': 347} | {'precision': 0.9201101928374655, 'recall': 0.962536023054755, 'f1': 0.9408450704225352, 'number': 347} | {'precision': 0.991304347826087, 'recall': 0.9855907780979827, 'f1': 0.9884393063583815, 'number': 347} | {'precision': 0.8397435897435898, 'recall': 0.7550432276657061, 'f1': 0.795144157814871, 'number': 347} | 0.9172 | 0.9099 | 0.9136 | 0.9936 |
63
+ | 0.0155 | 7.0 | 280 | 0.0290 | {'precision': 0.9129213483146067, 'recall': 0.9365994236311239, 'f1': 0.9246088193456615, 'number': 347} | {'precision': 0.9155313351498637, 'recall': 0.968299711815562, 'f1': 0.9411764705882353, 'number': 347} | {'precision': 0.9855907780979827, 'recall': 0.9855907780979827, 'f1': 0.9855907780979827, 'number': 347} | {'precision': 0.8228882833787466, 'recall': 0.8703170028818443, 'f1': 0.8459383753501399, 'number': 347} | 0.9081 | 0.9402 | 0.9239 | 0.9940 |
64
+ | 0.0125 | 8.0 | 320 | 0.0322 | {'precision': 0.9103641456582633, 'recall': 0.9365994236311239, 'f1': 0.9232954545454545, 'number': 347} | {'precision': 0.9027027027027027, 'recall': 0.962536023054755, 'f1': 0.9316596931659692, 'number': 347} | {'precision': 0.9884393063583815, 'recall': 0.9855907780979827, 'f1': 0.987012987012987, 'number': 347} | {'precision': 0.8269230769230769, 'recall': 0.8674351585014409, 'f1': 0.8466947960618847, 'number': 347} | 0.9061 | 0.9380 | 0.9218 | 0.9936 |
65
+ | 0.0108 | 9.0 | 360 | 0.0302 | {'precision': 0.9078212290502793, 'recall': 0.9365994236311239, 'f1': 0.921985815602837, 'number': 347} | {'precision': 0.9205479452054794, 'recall': 0.968299711815562, 'f1': 0.9438202247191011, 'number': 347} | {'precision': 0.9884393063583815, 'recall': 0.9855907780979827, 'f1': 0.987012987012987, 'number': 347} | {'precision': 0.8386167146974063, 'recall': 0.8386167146974063, 'f1': 0.8386167146974063, 'number': 347} | 0.9138 | 0.9323 | 0.9230 | 0.9941 |
66
+ | 0.0099 | 10.0 | 400 | 0.0304 | {'precision': 0.8833333333333333, 'recall': 0.9164265129682997, 'f1': 0.8995756718528995, 'number': 347} | {'precision': 0.9226519337016574, 'recall': 0.962536023054755, 'f1': 0.9421720733427363, 'number': 347} | {'precision': 0.9884393063583815, 'recall': 0.9855907780979827, 'f1': 0.987012987012987, 'number': 347} | {'precision': 0.8472222222222222, 'recall': 0.8789625360230547, 'f1': 0.8628005657708628, 'number': 347} | 0.9097 | 0.9359 | 0.9226 | 0.9943 |
67
+ | 0.0079 | 11.0 | 440 | 0.0312 | {'precision': 0.9206798866855525, 'recall': 0.9365994236311239, 'f1': 0.9285714285714285, 'number': 347} | {'precision': 0.9203296703296703, 'recall': 0.9654178674351584, 'f1': 0.9423347398030942, 'number': 347} | {'precision': 0.9884393063583815, 'recall': 0.9855907780979827, 'f1': 0.987012987012987, 'number': 347} | {'precision': 0.8515406162464986, 'recall': 0.8760806916426513, 'f1': 0.8636363636363638, 'number': 347} | 0.9197 | 0.9409 | 0.9302 | 0.9945 |
68
+ | 0.0079 | 12.0 | 480 | 0.0322 | {'precision': 0.9101123595505618, 'recall': 0.9337175792507204, 'f1': 0.9217638691322901, 'number': 347} | {'precision': 0.9153005464480874, 'recall': 0.9654178674351584, 'f1': 0.9396914446002805, 'number': 347} | {'precision': 0.9884393063583815, 'recall': 0.9855907780979827, 'f1': 0.987012987012987, 'number': 347} | {'precision': 0.8310626702997275, 'recall': 0.8789625360230547, 'f1': 0.8543417366946778, 'number': 347} | 0.9101 | 0.9409 | 0.9253 | 0.9941 |
69
+ | 0.0068 | 13.0 | 520 | 0.0311 | {'precision': 0.9101123595505618, 'recall': 0.9337175792507204, 'f1': 0.9217638691322901, 'number': 347} | {'precision': 0.9226519337016574, 'recall': 0.962536023054755, 'f1': 0.9421720733427363, 'number': 347} | {'precision': 0.991304347826087, 'recall': 0.9855907780979827, 'f1': 0.9884393063583815, 'number': 347} | {'precision': 0.8467966573816156, 'recall': 0.8760806916426513, 'f1': 0.8611898016997168, 'number': 347} | 0.9170 | 0.9395 | 0.9281 | 0.9943 |
70
+ | 0.0068 | 14.0 | 560 | 0.0318 | {'precision': 0.9101123595505618, 'recall': 0.9337175792507204, 'f1': 0.9217638691322901, 'number': 347} | {'precision': 0.9201101928374655, 'recall': 0.962536023054755, 'f1': 0.9408450704225352, 'number': 347} | {'precision': 0.9884393063583815, 'recall': 0.9855907780979827, 'f1': 0.987012987012987, 'number': 347} | {'precision': 0.837465564738292, 'recall': 0.8760806916426513, 'f1': 0.8563380281690142, 'number': 347} | 0.9132 | 0.9395 | 0.9261 | 0.9943 |
71
+ | 0.0061 | 15.0 | 600 | 0.0320 | {'precision': 0.9154929577464789, 'recall': 0.9365994236311239, 'f1': 0.925925925925926, 'number': 347} | {'precision': 0.9228650137741047, 'recall': 0.9654178674351584, 'f1': 0.9436619718309859, 'number': 347} | {'precision': 0.9884393063583815, 'recall': 0.9855907780979827, 'f1': 0.987012987012987, 'number': 347} | {'precision': 0.8431372549019608, 'recall': 0.8674351585014409, 'f1': 0.8551136363636365, 'number': 347} | 0.9170 | 0.9388 | 0.9277 | 0.9942 |
72
+
73
+
74
+ ### Framework versions
75
+
76
+ - Transformers 4.28.0
77
+ - Pytorch 2.1.0+cu118
78
+ - Datasets 2.15.0
79
+ - Tokenizers 0.13.3
logs/events.out.tfevents.1700199796.cbb148bf7329.232.0 CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:0f5dfdd5bf99d8fd09d422577f303b4cc8862890dcc85b6e6ad84253e8d42053
3
- size 14191
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:72d231aa884352270e5ba12b019a072e2a4994343f9198de843b2d3095401d63
3
+ size 14545
preprocessor_config.json ADDED
@@ -0,0 +1,14 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "apply_ocr": true,
3
+ "do_resize": true,
4
+ "feature_extractor_type": "LayoutLMv2FeatureExtractor",
5
+ "image_processor_type": "LayoutLMv2ImageProcessor",
6
+ "ocr_lang": null,
7
+ "processor_class": "LayoutLMv2Processor",
8
+ "resample": 2,
9
+ "size": {
10
+ "height": 224,
11
+ "width": 224
12
+ },
13
+ "tesseract_config": ""
14
+ }
pytorch_model.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:ed12c7dd9f26b0d98f1b77e59afb6a0fb05f1c3f02c06c4d39dd92dda6ae00a3
3
  size 450614978
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e4eda31901935df3183304cde30e47cf5c0b1cdf43d00897aef78bd16e1dfa4c
3
  size 450614978
special_tokens_map.json ADDED
@@ -0,0 +1,7 @@
 
 
 
 
 
 
 
 
1
+ {
2
+ "cls_token": "[CLS]",
3
+ "mask_token": "[MASK]",
4
+ "pad_token": "[PAD]",
5
+ "sep_token": "[SEP]",
6
+ "unk_token": "[UNK]"
7
+ }
tokenizer.json ADDED
The diff for this file is too large to render. See raw diff
 
tokenizer_config.json ADDED
@@ -0,0 +1,38 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "additional_special_tokens": null,
3
+ "apply_ocr": false,
4
+ "clean_up_tokenization_spaces": true,
5
+ "cls_token": "[CLS]",
6
+ "cls_token_box": [
7
+ 0,
8
+ 0,
9
+ 0,
10
+ 0
11
+ ],
12
+ "do_basic_tokenize": true,
13
+ "do_lower_case": true,
14
+ "mask_token": "[MASK]",
15
+ "model_max_length": 512,
16
+ "never_split": null,
17
+ "only_label_first_subword": true,
18
+ "pad_token": "[PAD]",
19
+ "pad_token_box": [
20
+ 0,
21
+ 0,
22
+ 0,
23
+ 0
24
+ ],
25
+ "pad_token_label": -100,
26
+ "processor_class": "LayoutLMv2Processor",
27
+ "sep_token": "[SEP]",
28
+ "sep_token_box": [
29
+ 1000,
30
+ 1000,
31
+ 1000,
32
+ 1000
33
+ ],
34
+ "strip_accents": null,
35
+ "tokenize_chinese_chars": true,
36
+ "tokenizer_class": "LayoutLMv2Tokenizer",
37
+ "unk_token": "[UNK]"
38
+ }
vocab.txt ADDED
The diff for this file is too large to render. See raw diff