stulcrad commited on
Commit
113b6c9
·
verified ·
1 Parent(s): 7bec812

End of training

Browse files
README.md ADDED
@@ -0,0 +1,81 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ library_name: transformers
3
+ license: cc-by-nc-sa-4.0
4
+ base_model: stulcrad/Robeczech-PRETRAINED4-CERED3
5
+ tags:
6
+ - generated_from_trainer
7
+ datasets:
8
+ - generator
9
+ metrics:
10
+ - accuracy
11
+ model-index:
12
+ - name: Robeczech-PRETRAINED43-CERED2
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
+ # Robeczech-PRETRAINED43-CERED2
20
+
21
+ This model is a fine-tuned version of [stulcrad/Robeczech-PRETRAINED4-CERED3](https://huggingface.co/stulcrad/Robeczech-PRETRAINED4-CERED3) on the generator dataset.
22
+ It achieves the following results on the evaluation set:
23
+ - Loss: 1.1137
24
+ - Accuracy: 0.8807
25
+ - Micro Precision: 0.8807
26
+ - Micro Recall: 0.8807
27
+ - Micro F1: 0.8807
28
+ - Macro Precision: 0.8503
29
+ - Macro Recall: 0.8424
30
+ - Macro F1: 0.8426
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: 0.0001
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: 10
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.4181 | 1.0 | 15074 | 0.5844 | 0.8550 | 0.8550 | 0.8550 | 0.8550 | 0.8202 | 0.7858 | 0.7946 |
65
+ | 0.3727 | 2.0 | 30148 | 0.5572 | 0.8620 | 0.8620 | 0.8620 | 0.8620 | 0.8293 | 0.8058 | 0.8102 |
66
+ | 0.282 | 3.0 | 45222 | 0.6841 | 0.8567 | 0.8567 | 0.8567 | 0.8567 | 0.8174 | 0.8071 | 0.8010 |
67
+ | 0.2209 | 4.0 | 60296 | 0.6510 | 0.8672 | 0.8672 | 0.8672 | 0.8672 | 0.8171 | 0.8205 | 0.8132 |
68
+ | 0.1918 | 5.0 | 75370 | 0.7609 | 0.8665 | 0.8665 | 0.8665 | 0.8665 | 0.8254 | 0.8171 | 0.8162 |
69
+ | 0.13 | 6.0 | 90444 | 0.8197 | 0.8724 | 0.8724 | 0.8724 | 0.8724 | 0.8347 | 0.8345 | 0.8302 |
70
+ | 0.0959 | 7.0 | 105518 | 0.8901 | 0.8721 | 0.8721 | 0.8721 | 0.8721 | 0.8304 | 0.8256 | 0.8236 |
71
+ | 0.0799 | 8.0 | 120592 | 1.0162 | 0.8749 | 0.8749 | 0.8749 | 0.8749 | 0.8364 | 0.8361 | 0.8316 |
72
+ | 0.0454 | 9.0 | 135666 | 1.0664 | 0.8747 | 0.8747 | 0.8747 | 0.8747 | 0.8280 | 0.8363 | 0.8284 |
73
+ | 0.0274 | 10.0 | 150740 | 1.1455 | 0.8768 | 0.8768 | 0.8768 | 0.8768 | 0.8326 | 0.8369 | 0.8313 |
74
+
75
+
76
+ ### Framework versions
77
+
78
+ - Transformers 4.46.2
79
+ - Pytorch 2.5.1+cu124
80
+ - Datasets 3.1.0
81
+ - Tokenizers 0.20.3
added_tokens.json ADDED
@@ -0,0 +1,7 @@
 
 
 
 
 
 
 
 
1
+ {
2
+ "[unused1]": 51997,
3
+ "[unused2]": 51998,
4
+ "[unused3]": 51999,
5
+ "[unused4]": 52000,
6
+ "[unused5]": 52001
7
+ }
config.json ADDED
@@ -0,0 +1,160 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "stulcrad/Robeczech-PRETRAINED4-CERED3",
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
+ "gradient_checkpointing": false,
11
+ "hidden_act": "gelu",
12
+ "hidden_dropout_prob": 0.1,
13
+ "hidden_size": 768,
14
+ "id2label": {
15
+ "0": "P0 other",
16
+ "1": "P19 m\u00edsto narozen\u00ed",
17
+ "2": "P20 m\u00edsto \u00famrt\u00ed",
18
+ "3": "P21 pohlav\u00ed",
19
+ "4": "P22 otec",
20
+ "5": "P25 matka",
21
+ "6": "P26 cho\u0165",
22
+ "7": "P27 st\u00e1tn\u00ed ob\u010danstv\u00ed",
23
+ "8": "P36 hlavn\u00ed m\u011bsto",
24
+ "9": "P39 ve funkci",
25
+ "10": "P47 hrani\u010d\u00ed s",
26
+ "11": "P50 autor",
27
+ "12": "P53 rodina",
28
+ "13": "P54 \u010dlen sportovn\u00edho dru\u017estva",
29
+ "14": "P97 \u0161lechtick\u00fd titul",
30
+ "15": "P101 zam\u011b\u0159en\u00ed",
31
+ "16": "P102 stranick\u00e1 p\u0159\u00edslu\u0161nost",
32
+ "17": "P105 taxonomick\u00e9 za\u0159azen\u00ed",
33
+ "18": "P106 povol\u00e1n\u00ed",
34
+ "19": "P118 liga",
35
+ "20": "P131 nach\u00e1z\u00ed se v administrativn\u00ed jednotce",
36
+ "21": "P136 \u017e\u00e1nr",
37
+ "22": "P137 oper\u00e1tor",
38
+ "23": "P138 pojmenov\u00e1no po",
39
+ "24": "P140 vyzn\u00e1n\u00ed",
40
+ "25": "P150 ni\u017e\u0161\u00ed spr\u00e1vn\u00ed celky",
41
+ "26": "P155 p\u0159edchoz\u00ed",
42
+ "27": "P159 s\u00eddlo",
43
+ "28": "P161 hraje",
44
+ "29": "P171 nad\u0159azen\u00fd taxon",
45
+ "30": "P175 interpret",
46
+ "31": "P176 v\u00fdrobce",
47
+ "32": "P179 s\u00e9rie",
48
+ "33": "P206 u vodn\u00ed plochy",
49
+ "34": "P264 hudebn\u00ed vydavatelstv\u00ed",
50
+ "35": "P276 m\u00edsto",
51
+ "36": "P279 nadt\u0159\u00edda",
52
+ "37": "P360 seznam (\u010deho)",
53
+ "38": "P361 \u010d\u00e1st (\u010deho)",
54
+ "39": "P413 pozice hr\u00e1\u010de",
55
+ "40": "P425 obor tohoto povol\u00e1n\u00ed",
56
+ "41": "P460 \u00fadajn\u011b tot\u00e9\u017e co",
57
+ "42": "P461 protiklad",
58
+ "43": "P463 \u010dlen (\u010deho)",
59
+ "44": "P495 zem\u011b p\u016fvodu",
60
+ "45": "P527 skl\u00e1d\u00e1 se z",
61
+ "46": "P530 diplomatick\u00fd vztah",
62
+ "47": "P641 sport",
63
+ "48": "P669 ulice",
64
+ "49": "P706 oblast",
65
+ "50": "P708 diec\u00e9ze",
66
+ "51": "P734 p\u0159\u00edjmen\u00ed",
67
+ "52": "P735 rodn\u00e9 jm\u00e9no",
68
+ "53": "P740 m\u00edsto vzniku",
69
+ "54": "P800 d\u00edlo",
70
+ "55": "P921 hlavn\u00ed t\u00e9ma d\u00edla",
71
+ "56": "P974 p\u0159\u00edtok",
72
+ "57": "P1056 produkuje",
73
+ "58": "P1303 hudebn\u00ed n\u00e1stroj",
74
+ "59": "P1376 hlavn\u00ed s\u00eddlo \u010deho",
75
+ "60": "P1383 zahrnuje s\u00eddlo",
76
+ "61": "P1889 rozd\u00edln\u00e9 od",
77
+ "62": "P3373 sourozenec",
78
+ "63": "P4552 poho\u0159\u00ed"
79
+ },
80
+ "initializer_range": 0.02,
81
+ "intermediate_size": 3072,
82
+ "label2id": {
83
+ "P0 other": 0,
84
+ "P101 zam\u011b\u0159en\u00ed": 15,
85
+ "P102 stranick\u00e1 p\u0159\u00edslu\u0161nost": 16,
86
+ "P105 taxonomick\u00e9 za\u0159azen\u00ed": 17,
87
+ "P1056 produkuje": 57,
88
+ "P106 povol\u00e1n\u00ed": 18,
89
+ "P118 liga": 19,
90
+ "P1303 hudebn\u00ed n\u00e1stroj": 58,
91
+ "P131 nach\u00e1z\u00ed se v administrativn\u00ed jednotce": 20,
92
+ "P136 \u017e\u00e1nr": 21,
93
+ "P137 oper\u00e1tor": 22,
94
+ "P1376 hlavn\u00ed s\u00eddlo \u010deho": 59,
95
+ "P138 pojmenov\u00e1no po": 23,
96
+ "P1383 zahrnuje s\u00eddlo": 60,
97
+ "P140 vyzn\u00e1n\u00ed": 24,
98
+ "P150 ni\u017e\u0161\u00ed spr\u00e1vn\u00ed celky": 25,
99
+ "P155 p\u0159edchoz\u00ed": 26,
100
+ "P159 s\u00eddlo": 27,
101
+ "P161 hraje": 28,
102
+ "P171 nad\u0159azen\u00fd taxon": 29,
103
+ "P175 interpret": 30,
104
+ "P176 v\u00fdrobce": 31,
105
+ "P179 s\u00e9rie": 32,
106
+ "P1889 rozd\u00edln\u00e9 od": 61,
107
+ "P19 m\u00edsto narozen\u00ed": 1,
108
+ "P20 m\u00edsto \u00famrt\u00ed": 2,
109
+ "P206 u vodn\u00ed plochy": 33,
110
+ "P21 pohlav\u00ed": 3,
111
+ "P22 otec": 4,
112
+ "P25 matka": 5,
113
+ "P26 cho\u0165": 6,
114
+ "P264 hudebn\u00ed vydavatelstv\u00ed": 34,
115
+ "P27 st\u00e1tn\u00ed ob\u010danstv\u00ed": 7,
116
+ "P276 m\u00edsto": 35,
117
+ "P279 nadt\u0159\u00edda": 36,
118
+ "P3373 sourozenec": 62,
119
+ "P36 hlavn\u00ed m\u011bsto": 8,
120
+ "P360 seznam (\u010deho)": 37,
121
+ "P361 \u010d\u00e1st (\u010deho)": 38,
122
+ "P39 ve funkci": 9,
123
+ "P413 pozice hr\u00e1\u010de": 39,
124
+ "P425 obor tohoto povol\u00e1n\u00ed": 40,
125
+ "P4552 poho\u0159\u00ed": 63,
126
+ "P460 \u00fadajn\u011b tot\u00e9\u017e co": 41,
127
+ "P461 protiklad": 42,
128
+ "P463 \u010dlen (\u010deho)": 43,
129
+ "P47 hrani\u010d\u00ed s": 10,
130
+ "P495 zem\u011b p\u016fvodu": 44,
131
+ "P50 autor": 11,
132
+ "P527 skl\u00e1d\u00e1 se z": 45,
133
+ "P53 rodina": 12,
134
+ "P530 diplomatick\u00fd vztah": 46,
135
+ "P54 \u010dlen sportovn\u00edho dru\u017estva": 13,
136
+ "P641 sport": 47,
137
+ "P669 ulice": 48,
138
+ "P706 oblast": 49,
139
+ "P708 diec\u00e9ze": 50,
140
+ "P734 p\u0159\u00edjmen\u00ed": 51,
141
+ "P735 rodn\u00e9 jm\u00e9no": 52,
142
+ "P740 m\u00edsto vzniku": 53,
143
+ "P800 d\u00edlo": 54,
144
+ "P921 hlavn\u00ed t\u00e9ma d\u00edla": 55,
145
+ "P97 \u0161lechtick\u00fd titul": 14,
146
+ "P974 p\u0159\u00edtok": 56
147
+ },
148
+ "layer_norm_eps": 1e-05,
149
+ "max_position_embeddings": 514,
150
+ "model_type": "roberta",
151
+ "num_attention_heads": 12,
152
+ "num_hidden_layers": 12,
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": 51997
160
+ }
merges.txt ADDED
The diff for this file is too large to render. See raw diff
 
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:45d8f06f2b6a986b6d025a5c9b356695bdc8b3fda9edc7a1c9846b827e6255d2
3
+ size 504532408
runs/Apr04_09-16-35_dgx10/events.out.tfevents.1743750997.dgx10.2636948.2 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9b4d09f4f11ff5169d2c24c8794ac08cadadaa31be8264655c3a13736aa24f5d
3
+ size 81581
runs/Apr04_09-16-35_dgx10/events.out.tfevents.1743765966.dgx10.2636948.3 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:14f55ec080a5d5ddf2c7933b930b75a075e1fa717dad1d9f6dc2a0c89b37497f
3
+ size 757
special_tokens_map.json ADDED
@@ -0,0 +1,88 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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": {
40
+ "content": "[CLS]",
41
+ "lstrip": false,
42
+ "normalized": true,
43
+ "rstrip": false,
44
+ "single_word": false
45
+ },
46
+ "cls_token": {
47
+ "content": "[CLS]",
48
+ "lstrip": false,
49
+ "normalized": true,
50
+ "rstrip": false,
51
+ "single_word": false
52
+ },
53
+ "eos_token": {
54
+ "content": "[SEP]",
55
+ "lstrip": false,
56
+ "normalized": true,
57
+ "rstrip": false,
58
+ "single_word": false
59
+ },
60
+ "mask_token": {
61
+ "content": "[MASK]",
62
+ "lstrip": true,
63
+ "normalized": true,
64
+ "rstrip": false,
65
+ "single_word": false
66
+ },
67
+ "pad_token": {
68
+ "content": "[PAD]",
69
+ "lstrip": false,
70
+ "normalized": true,
71
+ "rstrip": false,
72
+ "single_word": false
73
+ },
74
+ "sep_token": {
75
+ "content": "[SEP]",
76
+ "lstrip": false,
77
+ "normalized": true,
78
+ "rstrip": false,
79
+ "single_word": false
80
+ },
81
+ "unk_token": {
82
+ "content": "[UNK]",
83
+ "lstrip": false,
84
+ "normalized": true,
85
+ "rstrip": false,
86
+ "single_word": false
87
+ }
88
+ }
tokenizer.json ADDED
The diff for this file is too large to render. See raw diff
 
tokenizer_config.json ADDED
@@ -0,0 +1,108 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "add_prefix_space": true,
3
+ "added_tokens_decoder": {
4
+ "0": {
5
+ "content": "[CLS]",
6
+ "lstrip": false,
7
+ "normalized": true,
8
+ "rstrip": false,
9
+ "single_word": false,
10
+ "special": true
11
+ },
12
+ "1": {
13
+ "content": "[PAD]",
14
+ "lstrip": false,
15
+ "normalized": true,
16
+ "rstrip": false,
17
+ "single_word": false,
18
+ "special": true
19
+ },
20
+ "2": {
21
+ "content": "[SEP]",
22
+ "lstrip": false,
23
+ "normalized": true,
24
+ "rstrip": false,
25
+ "single_word": false,
26
+ "special": true
27
+ },
28
+ "3": {
29
+ "content": "[UNK]",
30
+ "lstrip": false,
31
+ "normalized": true,
32
+ "rstrip": false,
33
+ "single_word": false,
34
+ "special": true
35
+ },
36
+ "51960": {
37
+ "content": "[MASK]",
38
+ "lstrip": true,
39
+ "normalized": true,
40
+ "rstrip": false,
41
+ "single_word": false,
42
+ "special": true
43
+ },
44
+ "51997": {
45
+ "content": "[unused1]",
46
+ "lstrip": false,
47
+ "normalized": false,
48
+ "rstrip": false,
49
+ "single_word": false,
50
+ "special": true
51
+ },
52
+ "51998": {
53
+ "content": "[unused2]",
54
+ "lstrip": false,
55
+ "normalized": false,
56
+ "rstrip": false,
57
+ "single_word": false,
58
+ "special": true
59
+ },
60
+ "51999": {
61
+ "content": "[unused3]",
62
+ "lstrip": false,
63
+ "normalized": false,
64
+ "rstrip": false,
65
+ "single_word": false,
66
+ "special": true
67
+ },
68
+ "52000": {
69
+ "content": "[unused4]",
70
+ "lstrip": false,
71
+ "normalized": false,
72
+ "rstrip": false,
73
+ "single_word": false,
74
+ "special": true
75
+ },
76
+ "52001": {
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": "[CLS]",
93
+ "clean_up_tokenization_spaces": false,
94
+ "cls_token": "[CLS]",
95
+ "eos_token": "[SEP]",
96
+ "errors": "replace",
97
+ "mask_token": "[MASK]",
98
+ "max_length": 512,
99
+ "model_max_length": 512,
100
+ "pad_token": "[PAD]",
101
+ "sep_token": "[SEP]",
102
+ "stride": 0,
103
+ "tokenizer_class": "RobertaTokenizer",
104
+ "trim_offsets": true,
105
+ "truncation_side": "right",
106
+ "truncation_strategy": "longest_first",
107
+ "unk_token": "[UNK]"
108
+ }
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6ba9e29f5bdfcb3ac2df1d9600b70ba9e6bfa1da73df3926b87c2c8adea29c9d
3
+ size 5368
vocab.json ADDED
The diff for this file is too large to render. See raw diff