agata-luk commited on
Commit
19c2b97
·
verified ·
1 Parent(s): 8421cb8

Training complete

Browse files
Files changed (7) hide show
  1. .gitattributes +1 -0
  2. README.md +63 -0
  3. config.json +106 -0
  4. model.safetensors +3 -0
  5. tokenizer.json +3 -0
  6. tokenizer_config.json +24 -0
  7. training_args.bin +3 -0
.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,63 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ library_name: transformers
3
+ license: mit
4
+ base_model: jhu-clsp/mmBERT-base
5
+ tags:
6
+ - multi_label_classification
7
+ - generated_from_trainer
8
+ model-index:
9
+ - name: finetuned_model_emotion_detection
10
+ results: []
11
+ ---
12
+
13
+ <!-- This model card has been generated automatically according to the information the Trainer had access to. You
14
+ should probably proofread and complete it, then remove this comment. -->
15
+
16
+ # finetuned_model_emotion_detection
17
+
18
+ This model is a fine-tuned version of [jhu-clsp/mmBERT-base](https://huggingface.co/jhu-clsp/mmBERT-base) on the None dataset.
19
+ It achieves the following results on the evaluation set:
20
+ - Loss: 0.3381
21
+ - F1 Macro: 0.5114
22
+
23
+ ## Model description
24
+
25
+ More information needed
26
+
27
+ ## Intended uses & limitations
28
+
29
+ More information needed
30
+
31
+ ## Training and evaluation data
32
+
33
+ More information needed
34
+
35
+ ## Training procedure
36
+
37
+ ### Training hyperparameters
38
+
39
+ The following hyperparameters were used during training:
40
+ - learning_rate: 5e-05
41
+ - train_batch_size: 16
42
+ - eval_batch_size: 16
43
+ - seed: 42
44
+ - optimizer: Use OptimizerNames.ADAMW_TORCH_FUSED with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
45
+ - lr_scheduler_type: linear
46
+ - lr_scheduler_warmup_steps: 0.1
47
+ - num_epochs: 3
48
+
49
+ ### Training results
50
+
51
+ | Training Loss | Epoch | Step | Validation Loss | F1 Macro |
52
+ |:-------------:|:-----:|:----:|:---------------:|:--------:|
53
+ | No log | 1.0 | 223 | 0.2807 | 0.4083 |
54
+ | No log | 2.0 | 446 | 0.2764 | 0.4648 |
55
+ | 0.2685 | 3.0 | 669 | 0.3381 | 0.5114 |
56
+
57
+
58
+ ### Framework versions
59
+
60
+ - Transformers 5.3.0
61
+ - Pytorch 2.10.0+cu128
62
+ - Datasets 4.7.0
63
+ - Tokenizers 0.22.2
config.json ADDED
@@ -0,0 +1,106 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "architectures": [
3
+ "ModernBertForSequenceClassification"
4
+ ],
5
+ "attention_bias": false,
6
+ "attention_dropout": 0.0,
7
+ "bos_token_id": 2,
8
+ "classifier_activation": "gelu",
9
+ "classifier_bias": false,
10
+ "classifier_dropout": 0.0,
11
+ "classifier_pooling": "mean",
12
+ "cls_token_id": 1,
13
+ "decoder_bias": true,
14
+ "deterministic_flash_attn": false,
15
+ "dtype": "float32",
16
+ "embedding_dropout": 0.0,
17
+ "eos_token_id": 1,
18
+ "global_attn_every_n_layers": 3,
19
+ "gradient_checkpointing": false,
20
+ "hidden_activation": "gelu",
21
+ "hidden_size": 768,
22
+ "id2label": {
23
+ "0": "LABEL_0",
24
+ "1": "LABEL_1",
25
+ "2": "LABEL_2",
26
+ "3": "LABEL_3",
27
+ "4": "LABEL_4",
28
+ "5": "LABEL_5",
29
+ "6": "LABEL_6",
30
+ "7": "LABEL_7",
31
+ "8": "LABEL_8",
32
+ "9": "LABEL_9",
33
+ "10": "LABEL_10"
34
+ },
35
+ "initializer_cutoff_factor": 2.0,
36
+ "initializer_range": 0.02,
37
+ "intermediate_size": 1152,
38
+ "label2id": {
39
+ "LABEL_0": 0,
40
+ "LABEL_1": 1,
41
+ "LABEL_10": 10,
42
+ "LABEL_2": 2,
43
+ "LABEL_3": 3,
44
+ "LABEL_4": 4,
45
+ "LABEL_5": 5,
46
+ "LABEL_6": 6,
47
+ "LABEL_7": 7,
48
+ "LABEL_8": 8,
49
+ "LABEL_9": 9
50
+ },
51
+ "layer_norm_eps": 1e-05,
52
+ "layer_types": [
53
+ "full_attention",
54
+ "sliding_attention",
55
+ "sliding_attention",
56
+ "full_attention",
57
+ "sliding_attention",
58
+ "sliding_attention",
59
+ "full_attention",
60
+ "sliding_attention",
61
+ "sliding_attention",
62
+ "full_attention",
63
+ "sliding_attention",
64
+ "sliding_attention",
65
+ "full_attention",
66
+ "sliding_attention",
67
+ "sliding_attention",
68
+ "full_attention",
69
+ "sliding_attention",
70
+ "sliding_attention",
71
+ "full_attention",
72
+ "sliding_attention",
73
+ "sliding_attention",
74
+ "full_attention"
75
+ ],
76
+ "local_attention": 128,
77
+ "mask_token_id": 4,
78
+ "max_position_embeddings": 8192,
79
+ "mlp_bias": false,
80
+ "mlp_dropout": 0.0,
81
+ "model_type": "modernbert",
82
+ "norm_bias": false,
83
+ "norm_eps": 1e-05,
84
+ "num_attention_heads": 12,
85
+ "num_hidden_layers": 22,
86
+ "pad_token_id": 0,
87
+ "position_embedding_type": "sans_pos",
88
+ "problem_type": "multi_label_classification",
89
+ "rope_parameters": {
90
+ "full_attention": {
91
+ "rope_theta": 160000,
92
+ "rope_type": "default"
93
+ },
94
+ "sliding_attention": {
95
+ "rope_theta": 160000,
96
+ "rope_type": "default"
97
+ }
98
+ },
99
+ "sep_token_id": 1,
100
+ "sparse_pred_ignore_index": -100,
101
+ "sparse_prediction": false,
102
+ "tie_word_embeddings": true,
103
+ "transformers_version": "5.3.0",
104
+ "use_cache": false,
105
+ "vocab_size": 256000
106
+ }
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:29e8e08c8fafb0f05744fd0177462dd009e697af75bbe0fa2de138fa61a71c6f
3
+ size 1230169116
tokenizer.json ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:aa6396b645325378a159ce20800b8fdf4d3724306a7d601d47de108272e5761c
3
+ size 34363453
tokenizer_config.json ADDED
@@ -0,0 +1,24 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "backend": "tokenizers",
3
+ "bos_token": "<bos>",
4
+ "clean_up_tokenization_spaces": false,
5
+ "cls_token": "<bos>",
6
+ "eos_token": "<eos>",
7
+ "extra_special_tokens": [
8
+ "<start_of_turn>",
9
+ "<end_of_turn>"
10
+ ],
11
+ "is_local": false,
12
+ "mask_token": "<mask>",
13
+ "model_input_names": [
14
+ "input_ids",
15
+ "attention_mask"
16
+ ],
17
+ "model_max_length": 8192,
18
+ "pad_token": "<pad>",
19
+ "padding_side": "right",
20
+ "sep_token": "<eos>",
21
+ "spaces_between_special_tokens": false,
22
+ "tokenizer_class": "TokenizersBackend",
23
+ "unk_token": "<unk>"
24
+ }
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5eeb5858bf4c59ad3bad9da2e09292751034313ad74f8cf109e964b528403dd3
3
+ size 5201