stulcrad commited on
Commit
7f151a9
·
verified ·
1 Parent(s): b813911

End of training

Browse files
README.md ADDED
@@ -0,0 +1,76 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ library_name: transformers
3
+ license: apache-2.0
4
+ base_model: bert-base-multilingual-cased
5
+ tags:
6
+ - generated_from_trainer
7
+ datasets:
8
+ - generator
9
+ metrics:
10
+ - accuracy
11
+ model-index:
12
+ - name: mBERT-2
13
+ results: []
14
+ ---
15
+
16
+ <!-- This model card has been generated automatically according to the information the Trainer had access to. You
17
+ should probably proofread and complete it, then remove this comment. -->
18
+
19
+ # mBERT-2
20
+
21
+ This model is a fine-tuned version of [bert-base-multilingual-cased](https://huggingface.co/bert-base-multilingual-cased) on the generator dataset.
22
+ It achieves the following results on the evaluation set:
23
+ - Loss: 0.9469
24
+ - Accuracy: 0.8845
25
+ - Micro Precision: 0.8845
26
+ - Micro Recall: 0.8845
27
+ - Micro F1: 0.8845
28
+ - Macro Precision: 0.8512
29
+ - Macro Recall: 0.8349
30
+ - Macro F1: 0.8398
31
+
32
+ ## Model description
33
+
34
+ More information needed
35
+
36
+ ## Intended uses & limitations
37
+
38
+ More information needed
39
+
40
+ ## Training and evaluation data
41
+
42
+ More information needed
43
+
44
+ ## Training procedure
45
+
46
+ ### Training hyperparameters
47
+
48
+ The following hyperparameters were used during training:
49
+ - learning_rate: 5e-05
50
+ - train_batch_size: 12
51
+ - eval_batch_size: 12
52
+ - seed: 42
53
+ - optimizer: Use adamw_torch with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
54
+ - lr_scheduler_type: linear
55
+ - num_epochs: 8
56
+
57
+ ### Training results
58
+
59
+ | Training Loss | Epoch | Step | Validation Loss | Accuracy | Micro Precision | Micro Recall | Micro F1 | Macro Precision | Macro Recall | Macro F1 |
60
+ |:-------------:|:-----:|:------:|:---------------:|:--------:|:---------------:|:------------:|:--------:|:---------------:|:------------:|:--------:|
61
+ | 0.6019 | 1.0 | 30148 | 0.5612 | 0.8398 | 0.8398 | 0.8398 | 0.8398 | 0.8271 | 0.7687 | 0.7761 |
62
+ | 0.4904 | 2.0 | 60296 | 0.5216 | 0.8564 | 0.8564 | 0.8564 | 0.8564 | 0.8172 | 0.8079 | 0.8032 |
63
+ | 0.3624 | 3.0 | 90444 | 0.5196 | 0.8656 | 0.8656 | 0.8656 | 0.8656 | 0.8374 | 0.8006 | 0.8094 |
64
+ | 0.3242 | 4.0 | 120592 | 0.5662 | 0.8720 | 0.8720 | 0.8720 | 0.8720 | 0.8468 | 0.8222 | 0.8275 |
65
+ | 0.2522 | 5.0 | 150740 | 0.6456 | 0.8717 | 0.8717 | 0.8717 | 0.8717 | 0.8329 | 0.8374 | 0.8298 |
66
+ | 0.1707 | 6.0 | 180888 | 0.7233 | 0.8764 | 0.8764 | 0.8764 | 0.8764 | 0.8442 | 0.8356 | 0.8343 |
67
+ | 0.103 | 7.0 | 211036 | 0.8474 | 0.8794 | 0.8794 | 0.8794 | 0.8794 | 0.8440 | 0.8403 | 0.8385 |
68
+ | 0.0646 | 8.0 | 241184 | 0.9117 | 0.8836 | 0.8836 | 0.8836 | 0.8836 | 0.8465 | 0.8455 | 0.8423 |
69
+
70
+
71
+ ### Framework versions
72
+
73
+ - Transformers 4.46.2
74
+ - Pytorch 2.5.1+cu124
75
+ - Datasets 3.1.0
76
+ - Tokenizers 0.20.3
config.json ADDED
@@ -0,0 +1,163 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "bert-base-multilingual-cased",
3
+ "architectures": [
4
+ "RelationExtractionModel"
5
+ ],
6
+ "attention_probs_dropout_prob": 0.1,
7
+ "classifier_dropout": null,
8
+ "directionality": "bidi",
9
+ "hidden_act": "gelu",
10
+ "hidden_dropout_prob": 0.1,
11
+ "hidden_size": 768,
12
+ "id2label": {
13
+ "0": "P0 other",
14
+ "1": "P19 m\u00edsto narozen\u00ed",
15
+ "2": "P20 m\u00edsto \u00famrt\u00ed",
16
+ "3": "P21 pohlav\u00ed",
17
+ "4": "P22 otec",
18
+ "5": "P25 matka",
19
+ "6": "P26 cho\u0165",
20
+ "7": "P27 st\u00e1tn\u00ed ob\u010danstv\u00ed",
21
+ "8": "P36 hlavn\u00ed m\u011bsto",
22
+ "9": "P39 ve funkci",
23
+ "10": "P47 hrani\u010d\u00ed s",
24
+ "11": "P50 autor",
25
+ "12": "P53 rodina",
26
+ "13": "P54 \u010dlen sportovn\u00edho dru\u017estva",
27
+ "14": "P97 \u0161lechtick\u00fd titul",
28
+ "15": "P101 zam\u011b\u0159en\u00ed",
29
+ "16": "P102 stranick\u00e1 p\u0159\u00edslu\u0161nost",
30
+ "17": "P105 taxonomick\u00e9 za\u0159azen\u00ed",
31
+ "18": "P106 povol\u00e1n\u00ed",
32
+ "19": "P118 liga",
33
+ "20": "P131 nach\u00e1z\u00ed se v administrativn\u00ed jednotce",
34
+ "21": "P136 \u017e\u00e1nr",
35
+ "22": "P137 oper\u00e1tor",
36
+ "23": "P138 pojmenov\u00e1no po",
37
+ "24": "P140 vyzn\u00e1n\u00ed",
38
+ "25": "P150 ni\u017e\u0161\u00ed spr\u00e1vn\u00ed celky",
39
+ "26": "P155 p\u0159edchoz\u00ed",
40
+ "27": "P159 s\u00eddlo",
41
+ "28": "P161 hraje",
42
+ "29": "P171 nad\u0159azen\u00fd taxon",
43
+ "30": "P175 interpret",
44
+ "31": "P176 v\u00fdrobce",
45
+ "32": "P179 s\u00e9rie",
46
+ "33": "P206 u vodn\u00ed plochy",
47
+ "34": "P264 hudebn\u00ed vydavatelstv\u00ed",
48
+ "35": "P276 m\u00edsto",
49
+ "36": "P279 nadt\u0159\u00edda",
50
+ "37": "P360 seznam (\u010deho)",
51
+ "38": "P361 \u010d\u00e1st (\u010deho)",
52
+ "39": "P413 pozice hr\u00e1\u010de",
53
+ "40": "P425 obor tohoto povol\u00e1n\u00ed",
54
+ "41": "P460 \u00fadajn\u011b tot\u00e9\u017e co",
55
+ "42": "P461 protiklad",
56
+ "43": "P463 \u010dlen (\u010deho)",
57
+ "44": "P495 zem\u011b p\u016fvodu",
58
+ "45": "P527 skl\u00e1d\u00e1 se z",
59
+ "46": "P530 diplomatick\u00fd vztah",
60
+ "47": "P641 sport",
61
+ "48": "P669 ulice",
62
+ "49": "P706 oblast",
63
+ "50": "P708 diec\u00e9ze",
64
+ "51": "P734 p\u0159\u00edjmen\u00ed",
65
+ "52": "P735 rodn\u00e9 jm\u00e9no",
66
+ "53": "P740 m\u00edsto vzniku",
67
+ "54": "P800 d\u00edlo",
68
+ "55": "P921 hlavn\u00ed t\u00e9ma d\u00edla",
69
+ "56": "P974 p\u0159\u00edtok",
70
+ "57": "P1056 produkuje",
71
+ "58": "P1303 hudebn\u00ed n\u00e1stroj",
72
+ "59": "P1376 hlavn\u00ed s\u00eddlo \u010deho",
73
+ "60": "P1383 zahrnuje s\u00eddlo",
74
+ "61": "P1889 rozd\u00edln\u00e9 od",
75
+ "62": "P3373 sourozenec",
76
+ "63": "P4552 poho\u0159\u00ed"
77
+ },
78
+ "initializer_range": 0.02,
79
+ "intermediate_size": 3072,
80
+ "label2id": {
81
+ "P0 other": 0,
82
+ "P101 zam\u011b\u0159en\u00ed": 15,
83
+ "P102 stranick\u00e1 p\u0159\u00edslu\u0161nost": 16,
84
+ "P105 taxonomick\u00e9 za\u0159azen\u00ed": 17,
85
+ "P1056 produkuje": 57,
86
+ "P106 povol\u00e1n\u00ed": 18,
87
+ "P118 liga": 19,
88
+ "P1303 hudebn\u00ed n\u00e1stroj": 58,
89
+ "P131 nach\u00e1z\u00ed se v administrativn\u00ed jednotce": 20,
90
+ "P136 \u017e\u00e1nr": 21,
91
+ "P137 oper\u00e1tor": 22,
92
+ "P1376 hlavn\u00ed s\u00eddlo \u010deho": 59,
93
+ "P138 pojmenov\u00e1no po": 23,
94
+ "P1383 zahrnuje s\u00eddlo": 60,
95
+ "P140 vyzn\u00e1n\u00ed": 24,
96
+ "P150 ni\u017e\u0161\u00ed spr\u00e1vn\u00ed celky": 25,
97
+ "P155 p\u0159edchoz\u00ed": 26,
98
+ "P159 s\u00eddlo": 27,
99
+ "P161 hraje": 28,
100
+ "P171 nad\u0159azen\u00fd taxon": 29,
101
+ "P175 interpret": 30,
102
+ "P176 v\u00fdrobce": 31,
103
+ "P179 s\u00e9rie": 32,
104
+ "P1889 rozd\u00edln\u00e9 od": 61,
105
+ "P19 m\u00edsto narozen\u00ed": 1,
106
+ "P20 m\u00edsto \u00famrt\u00ed": 2,
107
+ "P206 u vodn\u00ed plochy": 33,
108
+ "P21 pohlav\u00ed": 3,
109
+ "P22 otec": 4,
110
+ "P25 matka": 5,
111
+ "P26 cho\u0165": 6,
112
+ "P264 hudebn\u00ed vydavatelstv\u00ed": 34,
113
+ "P27 st\u00e1tn\u00ed ob\u010danstv\u00ed": 7,
114
+ "P276 m\u00edsto": 35,
115
+ "P279 nadt\u0159\u00edda": 36,
116
+ "P3373 sourozenec": 62,
117
+ "P36 hlavn\u00ed m\u011bsto": 8,
118
+ "P360 seznam (\u010deho)": 37,
119
+ "P361 \u010d\u00e1st (\u010deho)": 38,
120
+ "P39 ve funkci": 9,
121
+ "P413 pozice hr\u00e1\u010de": 39,
122
+ "P425 obor tohoto povol\u00e1n\u00ed": 40,
123
+ "P4552 poho\u0159\u00ed": 63,
124
+ "P460 \u00fadajn\u011b tot\u00e9\u017e co": 41,
125
+ "P461 protiklad": 42,
126
+ "P463 \u010dlen (\u010deho)": 43,
127
+ "P47 hrani\u010d\u00ed s": 10,
128
+ "P495 zem\u011b p\u016fvodu": 44,
129
+ "P50 autor": 11,
130
+ "P527 skl\u00e1d\u00e1 se z": 45,
131
+ "P53 rodina": 12,
132
+ "P530 diplomatick\u00fd vztah": 46,
133
+ "P54 \u010dlen sportovn\u00edho dru\u017estva": 13,
134
+ "P641 sport": 47,
135
+ "P669 ulice": 48,
136
+ "P706 oblast": 49,
137
+ "P708 diec\u00e9ze": 50,
138
+ "P734 p\u0159\u00edjmen\u00ed": 51,
139
+ "P735 rodn\u00e9 jm\u00e9no": 52,
140
+ "P740 m\u00edsto vzniku": 53,
141
+ "P800 d\u00edlo": 54,
142
+ "P921 hlavn\u00ed t\u00e9ma d\u00edla": 55,
143
+ "P97 \u0161lechtick\u00fd titul": 14,
144
+ "P974 p\u0159\u00edtok": 56
145
+ },
146
+ "layer_norm_eps": 1e-12,
147
+ "max_position_embeddings": 512,
148
+ "model_type": "bert",
149
+ "num_attention_heads": 12,
150
+ "num_hidden_layers": 12,
151
+ "pad_token_id": 0,
152
+ "pooler_fc_size": 768,
153
+ "pooler_num_attention_heads": 12,
154
+ "pooler_num_fc_layers": 3,
155
+ "pooler_size_per_head": 128,
156
+ "pooler_type": "first_token_transform",
157
+ "position_embedding_type": "absolute",
158
+ "torch_dtype": "float32",
159
+ "transformers_version": "4.46.2",
160
+ "type_vocab_size": 2,
161
+ "use_cache": true,
162
+ "vocab_size": 119547
163
+ }
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2a002871ebaf57f1c843e507efc636586c9c042cf2cfa14d4d74f2f5c296c49f
3
+ size 712027576
runs/Dec20_14-09-14_n27/events.out.tfevents.1734700178.n27.269776.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:17ac176c427438cdea8947a4d48851b5a549af324f09d874c56e41cc8deca72f
3
+ size 9203
runs/Dec20_14-11-04_n27/events.out.tfevents.1734700266.n27.269776.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2763b1e5231b4a8fc9976ce10704af2672ac92b03d6fcf606b6bf3120d291c99
3
+ size 9203
runs/Dec20_14-26-53_g08/events.out.tfevents.1734701215.g08.3246119.4 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:65862eb8e52d66138cacd087a86f7a184e08bf9fa917abe92de695456408f17d
3
+ size 60452
runs/Feb03_16-51-14_dgx10/events.out.tfevents.1738597881.dgx10.76300.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:981d9055f04294b4eb124b24c062f389b30aa59176ade1e982cbfbf3aed263e4
3
+ size 119149
runs/Feb03_16-51-14_dgx10/events.out.tfevents.1738612513.dgx10.76300.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ef5907a25078a8cd504cc207ff199895ee9e82eae9f64033717d5c70696c409e
3
+ size 757
special_tokens_map.json ADDED
@@ -0,0 +1,44 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "additional_special_tokens": [
3
+ {
4
+ "content": "[unused1]",
5
+ "lstrip": false,
6
+ "normalized": false,
7
+ "rstrip": false,
8
+ "single_word": false
9
+ },
10
+ {
11
+ "content": "[unused2]",
12
+ "lstrip": false,
13
+ "normalized": false,
14
+ "rstrip": false,
15
+ "single_word": false
16
+ },
17
+ {
18
+ "content": "[unused3]",
19
+ "lstrip": false,
20
+ "normalized": false,
21
+ "rstrip": false,
22
+ "single_word": false
23
+ },
24
+ {
25
+ "content": "[unused4]",
26
+ "lstrip": false,
27
+ "normalized": false,
28
+ "rstrip": false,
29
+ "single_word": false
30
+ },
31
+ {
32
+ "content": "[unused5]",
33
+ "lstrip": false,
34
+ "normalized": false,
35
+ "rstrip": false,
36
+ "single_word": false
37
+ }
38
+ ],
39
+ "cls_token": "[CLS]",
40
+ "mask_token": "[MASK]",
41
+ "pad_token": "[PAD]",
42
+ "sep_token": "[SEP]",
43
+ "unk_token": "[UNK]"
44
+ }
tokenizer.json ADDED
The diff for this file is too large to render. See raw diff
 
tokenizer_config.json ADDED
@@ -0,0 +1,102 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "added_tokens_decoder": {
3
+ "0": {
4
+ "content": "[PAD]",
5
+ "lstrip": false,
6
+ "normalized": false,
7
+ "rstrip": false,
8
+ "single_word": false,
9
+ "special": true
10
+ },
11
+ "1": {
12
+ "content": "[unused1]",
13
+ "lstrip": false,
14
+ "normalized": false,
15
+ "rstrip": false,
16
+ "single_word": false,
17
+ "special": true
18
+ },
19
+ "2": {
20
+ "content": "[unused2]",
21
+ "lstrip": false,
22
+ "normalized": false,
23
+ "rstrip": false,
24
+ "single_word": false,
25
+ "special": true
26
+ },
27
+ "3": {
28
+ "content": "[unused3]",
29
+ "lstrip": false,
30
+ "normalized": false,
31
+ "rstrip": false,
32
+ "single_word": false,
33
+ "special": true
34
+ },
35
+ "4": {
36
+ "content": "[unused4]",
37
+ "lstrip": false,
38
+ "normalized": false,
39
+ "rstrip": false,
40
+ "single_word": false,
41
+ "special": true
42
+ },
43
+ "5": {
44
+ "content": "[unused5]",
45
+ "lstrip": false,
46
+ "normalized": false,
47
+ "rstrip": false,
48
+ "single_word": false,
49
+ "special": true
50
+ },
51
+ "100": {
52
+ "content": "[UNK]",
53
+ "lstrip": false,
54
+ "normalized": false,
55
+ "rstrip": false,
56
+ "single_word": false,
57
+ "special": true
58
+ },
59
+ "101": {
60
+ "content": "[CLS]",
61
+ "lstrip": false,
62
+ "normalized": false,
63
+ "rstrip": false,
64
+ "single_word": false,
65
+ "special": true
66
+ },
67
+ "102": {
68
+ "content": "[SEP]",
69
+ "lstrip": false,
70
+ "normalized": false,
71
+ "rstrip": false,
72
+ "single_word": false,
73
+ "special": true
74
+ },
75
+ "103": {
76
+ "content": "[MASK]",
77
+ "lstrip": false,
78
+ "normalized": false,
79
+ "rstrip": false,
80
+ "single_word": false,
81
+ "special": true
82
+ }
83
+ },
84
+ "additional_special_tokens": [
85
+ "[unused1]",
86
+ "[unused2]",
87
+ "[unused3]",
88
+ "[unused4]",
89
+ "[unused5]"
90
+ ],
91
+ "clean_up_tokenization_spaces": false,
92
+ "cls_token": "[CLS]",
93
+ "do_lower_case": false,
94
+ "mask_token": "[MASK]",
95
+ "model_max_length": 512,
96
+ "pad_token": "[PAD]",
97
+ "sep_token": "[SEP]",
98
+ "strip_accents": null,
99
+ "tokenize_chinese_chars": true,
100
+ "tokenizer_class": "BertTokenizer",
101
+ "unk_token": "[UNK]"
102
+ }
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e2aa2204c2df7891c355527756f7f876fcc11529368cf074ad96779e102d4328
3
+ size 5304
vocab.txt ADDED
The diff for this file is too large to render. See raw diff