contemmcm commited on
Commit
3c8b24d
·
verified ·
1 Parent(s): e182e83

End of training

Browse files
README.md ADDED
@@ -0,0 +1,80 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ library_name: transformers
3
+ license: mit
4
+ base_model: FacebookAI/roberta-base
5
+ tags:
6
+ - generated_from_trainer
7
+ metrics:
8
+ - accuracy
9
+ model-index:
10
+ - name: f0e34afffe85e814b4e41abc80263fca
11
+ results: []
12
+ ---
13
+
14
+ <!-- This model card has been generated automatically according to the information the Trainer had access to. You
15
+ should probably proofread and complete it, then remove this comment. -->
16
+
17
+ # f0e34afffe85e814b4e41abc80263fca
18
+
19
+ This model is a fine-tuned version of [FacebookAI/roberta-base](https://huggingface.co/FacebookAI/roberta-base) on the contemmcm/hate-speech-and-offensive-language dataset.
20
+ It achieves the following results on the evaluation set:
21
+ - Loss: 0.3189
22
+ - Data Size: 1.0
23
+ - Epoch Runtime: 49.4083
24
+ - Accuracy: 0.9115
25
+ - F1 Macro: 0.7587
26
+
27
+ ## Model description
28
+
29
+ More information needed
30
+
31
+ ## Intended uses & limitations
32
+
33
+ More information needed
34
+
35
+ ## Training and evaluation data
36
+
37
+ More information needed
38
+
39
+ ## Training procedure
40
+
41
+ ### Training hyperparameters
42
+
43
+ The following hyperparameters were used during training:
44
+ - learning_rate: 5e-05
45
+ - train_batch_size: 8
46
+ - eval_batch_size: 8
47
+ - seed: 42
48
+ - distributed_type: multi-GPU
49
+ - num_devices: 4
50
+ - total_train_batch_size: 32
51
+ - total_eval_batch_size: 32
52
+ - optimizer: Use adamw_torch with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
53
+ - lr_scheduler_type: constant
54
+ - num_epochs: 50
55
+
56
+ ### Training results
57
+
58
+ | Training Loss | Epoch | Step | Validation Loss | Data Size | Epoch Runtime | Accuracy | F1 Macro |
59
+ |:-------------:|:-----:|:----:|:---------------:|:---------:|:-------------:|:--------:|:--------:|
60
+ | No log | 0 | 0 | 1.2039 | 0 | 4.1789 | 0.0601 | 0.0378 |
61
+ | No log | 1 | 619 | 0.6824 | 0.0078 | 4.7944 | 0.7672 | 0.2894 |
62
+ | No log | 2 | 1238 | 0.6033 | 0.0156 | 5.1378 | 0.7672 | 0.2894 |
63
+ | 0.0148 | 3 | 1857 | 0.3472 | 0.0312 | 5.9556 | 0.8904 | 0.5871 |
64
+ | 0.0148 | 4 | 2476 | 0.2996 | 0.0625 | 7.7942 | 0.9046 | 0.6062 |
65
+ | 0.3137 | 5 | 3095 | 0.2770 | 0.125 | 10.1123 | 0.9065 | 0.7305 |
66
+ | 0.0257 | 6 | 3714 | 0.2918 | 0.25 | 15.7152 | 0.9054 | 0.6093 |
67
+ | 0.2836 | 7 | 4333 | 0.3122 | 0.5 | 27.3986 | 0.8981 | 0.7566 |
68
+ | 0.2616 | 8.0 | 4952 | 0.2436 | 1.0 | 50.3288 | 0.9164 | 0.6985 |
69
+ | 0.2095 | 9.0 | 5571 | 0.3629 | 1.0 | 49.8399 | 0.8843 | 0.6892 |
70
+ | 0.2033 | 10.0 | 6190 | 0.2972 | 1.0 | 49.5511 | 0.9150 | 0.7817 |
71
+ | 0.2164 | 11.0 | 6809 | 0.2992 | 1.0 | 50.8395 | 0.9093 | 0.7611 |
72
+ | 0.1258 | 12.0 | 7428 | 0.3189 | 1.0 | 49.4083 | 0.9115 | 0.7587 |
73
+
74
+
75
+ ### Framework versions
76
+
77
+ - Transformers 4.57.0
78
+ - Pytorch 2.8.0+cu128
79
+ - Datasets 4.3.0
80
+ - Tokenizers 0.22.1
config.json ADDED
@@ -0,0 +1,37 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "architectures": [
3
+ "RobertaForSequenceClassification"
4
+ ],
5
+ "attention_probs_dropout_prob": 0.1,
6
+ "bos_token_id": 0,
7
+ "classifier_dropout": null,
8
+ "dtype": "float32",
9
+ "eos_token_id": 2,
10
+ "hidden_act": "gelu",
11
+ "hidden_dropout_prob": 0.1,
12
+ "hidden_size": 768,
13
+ "id2label": {
14
+ "0": "LABEL_0",
15
+ "1": "LABEL_1",
16
+ "2": "LABEL_2"
17
+ },
18
+ "initializer_range": 0.02,
19
+ "intermediate_size": 3072,
20
+ "label2id": {
21
+ "LABEL_0": 0,
22
+ "LABEL_1": 1,
23
+ "LABEL_2": 2
24
+ },
25
+ "layer_norm_eps": 1e-05,
26
+ "max_position_embeddings": 514,
27
+ "model_type": "roberta",
28
+ "num_attention_heads": 12,
29
+ "num_hidden_layers": 12,
30
+ "pad_token_id": 1,
31
+ "position_embedding_type": "absolute",
32
+ "problem_type": "single_label_classification",
33
+ "transformers_version": "4.57.0",
34
+ "type_vocab_size": 1,
35
+ "use_cache": true,
36
+ "vocab_size": 50265
37
+ }
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:0723cc21df5888420f6f2434df8d6eb05ffe61ad71d253cb527dda5b1625690f
3
+ size 498615900
special_tokens_map.json ADDED
@@ -0,0 +1,15 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "bos_token": "<s>",
3
+ "cls_token": "<s>",
4
+ "eos_token": "</s>",
5
+ "mask_token": {
6
+ "content": "<mask>",
7
+ "lstrip": true,
8
+ "normalized": false,
9
+ "rstrip": false,
10
+ "single_word": false
11
+ },
12
+ "pad_token": "<pad>",
13
+ "sep_token": "</s>",
14
+ "unk_token": "<unk>"
15
+ }
tokenizer.json ADDED
The diff for this file is too large to render. See raw diff
 
tokenizer_config.json ADDED
@@ -0,0 +1,58 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "add_prefix_space": false,
3
+ "added_tokens_decoder": {
4
+ "0": {
5
+ "content": "<s>",
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": "</s>",
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
+ "50264": {
37
+ "content": "<mask>",
38
+ "lstrip": true,
39
+ "normalized": false,
40
+ "rstrip": false,
41
+ "single_word": false,
42
+ "special": true
43
+ }
44
+ },
45
+ "bos_token": "<s>",
46
+ "clean_up_tokenization_spaces": false,
47
+ "cls_token": "<s>",
48
+ "eos_token": "</s>",
49
+ "errors": "replace",
50
+ "extra_special_tokens": {},
51
+ "mask_token": "<mask>",
52
+ "model_max_length": 512,
53
+ "pad_token": "<pad>",
54
+ "sep_token": "</s>",
55
+ "tokenizer_class": "RobertaTokenizer",
56
+ "trim_offsets": true,
57
+ "unk_token": "<unk>"
58
+ }
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a62d1bb8299de9929e64ce9c597e94aa5e923e0f483199fdb7bbca8120efe792
3
+ size 5969
vocab.json ADDED
The diff for this file is too large to render. See raw diff