stulcrad commited on
Commit
905f979
·
verified ·
1 Parent(s): f53ad8b

End of training

Browse files
.gitattributes CHANGED
@@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
 
 
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
36
+ tokenizer.json filter=lfs diff=lfs merge=lfs -text
README.md ADDED
@@ -0,0 +1,76 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ library_name: transformers
3
+ license: mit
4
+ base_model: xlm-roberta-large
5
+ tags:
6
+ - generated_from_trainer
7
+ datasets:
8
+ - generator
9
+ metrics:
10
+ - accuracy
11
+ model-index:
12
+ - name: XLM-RoBERTa-1
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
+ # XLM-RoBERTa-1
20
+
21
+ This model is a fine-tuned version of [xlm-roberta-large](https://huggingface.co/xlm-roberta-large) on the generator dataset.
22
+ It achieves the following results on the evaluation set:
23
+ - Loss: 0.4935
24
+ - Accuracy: 0.9044
25
+ - Micro Precision: 0.9044
26
+ - Micro Recall: 0.9044
27
+ - Micro F1: 0.9044
28
+ - Macro Precision: 0.8822
29
+ - Macro Recall: 0.8527
30
+ - Macro F1: 0.8599
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: 1e-05
50
+ - train_batch_size: 12
51
+ - eval_batch_size: 12
52
+ - seed: 42
53
+ - gradient_accumulation_steps: 2
54
+ - total_train_batch_size: 24
55
+ - optimizer: Use adamw_torch with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
56
+ - lr_scheduler_type: linear
57
+ - lr_scheduler_warmup_steps: 500
58
+ - num_epochs: 5
59
+
60
+ ### Training results
61
+
62
+ | Training Loss | Epoch | Step | Validation Loss | Accuracy | Micro Precision | Micro Recall | Micro F1 | Macro Precision | Macro Recall | Macro F1 |
63
+ |:-------------:|:------:|:------:|:---------------:|:--------:|:---------------:|:------------:|:--------:|:---------------:|:------------:|:--------:|
64
+ | 0.3304 | 1.0000 | 40166 | 0.3911 | 0.8883 | 0.8883 | 0.8883 | 0.8883 | 0.8651 | 0.8285 | 0.8359 |
65
+ | 0.2376 | 2.0 | 80333 | 0.3998 | 0.9014 | 0.9014 | 0.9014 | 0.9014 | 0.8714 | 0.8480 | 0.8523 |
66
+ | 0.1733 | 3.0000 | 120499 | 0.4297 | 0.9059 | 0.9059 | 0.9059 | 0.9059 | 0.8718 | 0.8607 | 0.8610 |
67
+ | 0.1268 | 4.0 | 160666 | 0.4822 | 0.9088 | 0.9088 | 0.9088 | 0.9088 | 0.8807 | 0.8608 | 0.8649 |
68
+ | 0.1038 | 4.9999 | 200830 | 0.5242 | 0.9087 | 0.9087 | 0.9087 | 0.9087 | 0.8753 | 0.8646 | 0.8660 |
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,160 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "xlm-roberta-large",
3
+ "architectures": [
4
+ "RelationExtractionModel"
5
+ ],
6
+ "attention_probs_dropout_prob": 0.1,
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": 1024,
13
+ "id2label": {
14
+ "0": "P0 other",
15
+ "1": "P19 m\u00edsto narozen\u00ed",
16
+ "2": "P20 m\u00edsto \u00famrt\u00ed",
17
+ "3": "P21 pohlav\u00ed",
18
+ "4": "P22 otec",
19
+ "5": "P25 matka",
20
+ "6": "P26 cho\u0165",
21
+ "7": "P27 st\u00e1tn\u00ed ob\u010danstv\u00ed",
22
+ "8": "P36 hlavn\u00ed m\u011bsto",
23
+ "9": "P39 ve funkci",
24
+ "10": "P47 hrani\u010d\u00ed s",
25
+ "11": "P50 autor",
26
+ "12": "P53 rodina",
27
+ "13": "P54 \u010dlen sportovn\u00edho dru\u017estva",
28
+ "14": "P97 \u0161lechtick\u00fd titul",
29
+ "15": "P101 zam\u011b\u0159en\u00ed",
30
+ "16": "P102 stranick\u00e1 p\u0159\u00edslu\u0161nost",
31
+ "17": "P105 taxonomick\u00e9 za\u0159azen\u00ed",
32
+ "18": "P106 povol\u00e1n\u00ed",
33
+ "19": "P118 liga",
34
+ "20": "P131 nach\u00e1z\u00ed se v administrativn\u00ed jednotce",
35
+ "21": "P136 \u017e\u00e1nr",
36
+ "22": "P137 oper\u00e1tor",
37
+ "23": "P138 pojmenov\u00e1no po",
38
+ "24": "P140 vyzn\u00e1n\u00ed",
39
+ "25": "P150 ni\u017e\u0161\u00ed spr\u00e1vn\u00ed celky",
40
+ "26": "P155 p\u0159edchoz\u00ed",
41
+ "27": "P159 s\u00eddlo",
42
+ "28": "P161 hraje",
43
+ "29": "P171 nad\u0159azen\u00fd taxon",
44
+ "30": "P175 interpret",
45
+ "31": "P176 v\u00fdrobce",
46
+ "32": "P179 s\u00e9rie",
47
+ "33": "P206 u vodn\u00ed plochy",
48
+ "34": "P264 hudebn\u00ed vydavatelstv\u00ed",
49
+ "35": "P276 m\u00edsto",
50
+ "36": "P279 nadt\u0159\u00edda",
51
+ "37": "P360 seznam (\u010deho)",
52
+ "38": "P361 \u010d\u00e1st (\u010deho)",
53
+ "39": "P413 pozice hr\u00e1\u010de",
54
+ "40": "P425 obor tohoto povol\u00e1n\u00ed",
55
+ "41": "P460 \u00fadajn\u011b tot\u00e9\u017e co",
56
+ "42": "P461 protiklad",
57
+ "43": "P463 \u010dlen (\u010deho)",
58
+ "44": "P495 zem\u011b p\u016fvodu",
59
+ "45": "P527 skl\u00e1d\u00e1 se z",
60
+ "46": "P530 diplomatick\u00fd vztah",
61
+ "47": "P641 sport",
62
+ "48": "P669 ulice",
63
+ "49": "P706 oblast",
64
+ "50": "P708 diec\u00e9ze",
65
+ "51": "P734 p\u0159\u00edjmen\u00ed",
66
+ "52": "P735 rodn\u00e9 jm\u00e9no",
67
+ "53": "P740 m\u00edsto vzniku",
68
+ "54": "P800 d\u00edlo",
69
+ "55": "P921 hlavn\u00ed t\u00e9ma d\u00edla",
70
+ "56": "P974 p\u0159\u00edtok",
71
+ "57": "P1056 produkuje",
72
+ "58": "P1303 hudebn\u00ed n\u00e1stroj",
73
+ "59": "P1376 hlavn\u00ed s\u00eddlo \u010deho",
74
+ "60": "P1383 zahrnuje s\u00eddlo",
75
+ "61": "P1889 rozd\u00edln\u00e9 od",
76
+ "62": "P3373 sourozenec",
77
+ "63": "P4552 poho\u0159\u00ed"
78
+ },
79
+ "initializer_range": 0.02,
80
+ "intermediate_size": 4096,
81
+ "label2id": {
82
+ "P0 other": 0,
83
+ "P101 zam\u011b\u0159en\u00ed": 15,
84
+ "P102 stranick\u00e1 p\u0159\u00edslu\u0161nost": 16,
85
+ "P105 taxonomick\u00e9 za\u0159azen\u00ed": 17,
86
+ "P1056 produkuje": 57,
87
+ "P106 povol\u00e1n\u00ed": 18,
88
+ "P118 liga": 19,
89
+ "P1303 hudebn\u00ed n\u00e1stroj": 58,
90
+ "P131 nach\u00e1z\u00ed se v administrativn\u00ed jednotce": 20,
91
+ "P136 \u017e\u00e1nr": 21,
92
+ "P137 oper\u00e1tor": 22,
93
+ "P1376 hlavn\u00ed s\u00eddlo \u010deho": 59,
94
+ "P138 pojmenov\u00e1no po": 23,
95
+ "P1383 zahrnuje s\u00eddlo": 60,
96
+ "P140 vyzn\u00e1n\u00ed": 24,
97
+ "P150 ni\u017e\u0161\u00ed spr\u00e1vn\u00ed celky": 25,
98
+ "P155 p\u0159edchoz\u00ed": 26,
99
+ "P159 s\u00eddlo": 27,
100
+ "P161 hraje": 28,
101
+ "P171 nad\u0159azen\u00fd taxon": 29,
102
+ "P175 interpret": 30,
103
+ "P176 v\u00fdrobce": 31,
104
+ "P179 s\u00e9rie": 32,
105
+ "P1889 rozd\u00edln\u00e9 od": 61,
106
+ "P19 m\u00edsto narozen\u00ed": 1,
107
+ "P20 m\u00edsto \u00famrt\u00ed": 2,
108
+ "P206 u vodn\u00ed plochy": 33,
109
+ "P21 pohlav\u00ed": 3,
110
+ "P22 otec": 4,
111
+ "P25 matka": 5,
112
+ "P26 cho\u0165": 6,
113
+ "P264 hudebn\u00ed vydavatelstv\u00ed": 34,
114
+ "P27 st\u00e1tn\u00ed ob\u010danstv\u00ed": 7,
115
+ "P276 m\u00edsto": 35,
116
+ "P279 nadt\u0159\u00edda": 36,
117
+ "P3373 sourozenec": 62,
118
+ "P36 hlavn\u00ed m\u011bsto": 8,
119
+ "P360 seznam (\u010deho)": 37,
120
+ "P361 \u010d\u00e1st (\u010deho)": 38,
121
+ "P39 ve funkci": 9,
122
+ "P413 pozice hr\u00e1\u010de": 39,
123
+ "P425 obor tohoto povol\u00e1n\u00ed": 40,
124
+ "P4552 poho\u0159\u00ed": 63,
125
+ "P460 \u00fadajn\u011b tot\u00e9\u017e co": 41,
126
+ "P461 protiklad": 42,
127
+ "P463 \u010dlen (\u010deho)": 43,
128
+ "P47 hrani\u010d\u00ed s": 10,
129
+ "P495 zem\u011b p\u016fvodu": 44,
130
+ "P50 autor": 11,
131
+ "P527 skl\u00e1d\u00e1 se z": 45,
132
+ "P53 rodina": 12,
133
+ "P530 diplomatick\u00fd vztah": 46,
134
+ "P54 \u010dlen sportovn\u00edho dru\u017estva": 13,
135
+ "P641 sport": 47,
136
+ "P669 ulice": 48,
137
+ "P706 oblast": 49,
138
+ "P708 diec\u00e9ze": 50,
139
+ "P734 p\u0159\u00edjmen\u00ed": 51,
140
+ "P735 rodn\u00e9 jm\u00e9no": 52,
141
+ "P740 m\u00edsto vzniku": 53,
142
+ "P800 d\u00edlo": 54,
143
+ "P921 hlavn\u00ed t\u00e9ma d\u00edla": 55,
144
+ "P97 \u0161lechtick\u00fd titul": 14,
145
+ "P974 p\u0159\u00edtok": 56
146
+ },
147
+ "layer_norm_eps": 1e-05,
148
+ "max_position_embeddings": 514,
149
+ "model_type": "xlm-roberta",
150
+ "num_attention_heads": 16,
151
+ "num_hidden_layers": 24,
152
+ "output_past": true,
153
+ "pad_token_id": 1,
154
+ "position_embedding_type": "absolute",
155
+ "torch_dtype": "float32",
156
+ "transformers_version": "4.46.2",
157
+ "type_vocab_size": 1,
158
+ "use_cache": true,
159
+ "vocab_size": 250002
160
+ }
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:beb92342efa5ce3788ec5a2100c32f556cd32c6fea6e76ae777474e01fed902b
3
+ size 2240416848
runs/Mar06_01-39-41_dgx10/events.out.tfevents.1741221586.dgx10.2696097.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4bd472cd03ceafab66d9b213213d3df0a94df23c6f8016b1d310158c5bd6504a
3
+ size 9852
runs/Mar06_01-55-57_dgx10/events.out.tfevents.1741222561.dgx10.2700023.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:159acffbfb143f476b468d225d25e8c54e0bd9170c5d8b8629f330f57d46ed57
3
+ size 99643
runs/Mar06_01-55-57_dgx10/events.out.tfevents.1741297853.dgx10.2700023.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3b86b858d7334c092fcf94d4709237688bc7924a14e9e320ba716bfff8bf88d4
3
+ size 757
special_tokens_map.json ADDED
@@ -0,0 +1,52 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
+ "bos_token": "<s>",
40
+ "cls_token": "<s>",
41
+ "eos_token": "</s>",
42
+ "mask_token": {
43
+ "content": "<mask>",
44
+ "lstrip": true,
45
+ "normalized": false,
46
+ "rstrip": false,
47
+ "single_word": false
48
+ },
49
+ "pad_token": "<pad>",
50
+ "sep_token": "</s>",
51
+ "unk_token": "<unk>"
52
+ }
tokenizer.json ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:cd1ed561533c9d74649ec70d231aaad68ec693dd3e1821c5da8a69e59ce6a56e
3
+ size 17083762
tokenizer_config.json ADDED
@@ -0,0 +1,102 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "add_prefix_space": true,
3
+ "added_tokens_decoder": {
4
+ "0": {
5
+ "content": "<s>",
6
+ "lstrip": false,
7
+ "normalized": false,
8
+ "rstrip": false,
9
+ "single_word": false,
10
+ "special": true
11
+ },
12
+ "1": {
13
+ "content": "<pad>",
14
+ "lstrip": false,
15
+ "normalized": false,
16
+ "rstrip": false,
17
+ "single_word": false,
18
+ "special": true
19
+ },
20
+ "2": {
21
+ "content": "</s>",
22
+ "lstrip": false,
23
+ "normalized": false,
24
+ "rstrip": false,
25
+ "single_word": false,
26
+ "special": true
27
+ },
28
+ "3": {
29
+ "content": "<unk>",
30
+ "lstrip": false,
31
+ "normalized": false,
32
+ "rstrip": false,
33
+ "single_word": false,
34
+ "special": true
35
+ },
36
+ "250001": {
37
+ "content": "<mask>",
38
+ "lstrip": true,
39
+ "normalized": false,
40
+ "rstrip": false,
41
+ "single_word": false,
42
+ "special": true
43
+ },
44
+ "250002": {
45
+ "content": "[unused1]",
46
+ "lstrip": false,
47
+ "normalized": false,
48
+ "rstrip": false,
49
+ "single_word": false,
50
+ "special": true
51
+ },
52
+ "250003": {
53
+ "content": "[unused2]",
54
+ "lstrip": false,
55
+ "normalized": false,
56
+ "rstrip": false,
57
+ "single_word": false,
58
+ "special": true
59
+ },
60
+ "250004": {
61
+ "content": "[unused3]",
62
+ "lstrip": false,
63
+ "normalized": false,
64
+ "rstrip": false,
65
+ "single_word": false,
66
+ "special": true
67
+ },
68
+ "250005": {
69
+ "content": "[unused4]",
70
+ "lstrip": false,
71
+ "normalized": false,
72
+ "rstrip": false,
73
+ "single_word": false,
74
+ "special": true
75
+ },
76
+ "250006": {
77
+ "content": "[unused5]",
78
+ "lstrip": false,
79
+ "normalized": false,
80
+ "rstrip": false,
81
+ "single_word": false,
82
+ "special": true
83
+ }
84
+ },
85
+ "additional_special_tokens": [
86
+ "[unused1]",
87
+ "[unused2]",
88
+ "[unused3]",
89
+ "[unused4]",
90
+ "[unused5]"
91
+ ],
92
+ "bos_token": "<s>",
93
+ "clean_up_tokenization_spaces": false,
94
+ "cls_token": "<s>",
95
+ "eos_token": "</s>",
96
+ "mask_token": "<mask>",
97
+ "model_max_length": 512,
98
+ "pad_token": "<pad>",
99
+ "sep_token": "</s>",
100
+ "tokenizer_class": "XLMRobertaTokenizer",
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:b2476dc9ac594194dc0989969d790e0761fea836a1e19e5e237763dbf6ff5f5b
3
+ size 5304