MM2157 commited on
Commit
c880475
·
verified ·
1 Parent(s): 0f58bcd

Sussex_AI_marbert_arabic_sentiment_5F

Browse files
Files changed (4) hide show
  1. README.md +62 -0
  2. config.json +47 -0
  3. model.safetensors +3 -0
  4. training_args.bin +3 -0
README.md ADDED
@@ -0,0 +1,62 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ library_name: transformers
3
+ base_model: UBC-NLP/MARBERT
4
+ tags:
5
+ - generated_from_trainer
6
+ model-index:
7
+ - name: fold_4
8
+ results: []
9
+ ---
10
+
11
+ <!-- This model card has been generated automatically according to the information the Trainer had access to. You
12
+ should probably proofread and complete it, then remove this comment. -->
13
+
14
+ # fold_4
15
+
16
+ This model is a fine-tuned version of [UBC-NLP/MARBERT](https://huggingface.co/UBC-NLP/MARBERT) on the None dataset.
17
+ It achieves the following results on the evaluation set:
18
+ - Loss: 0.4025
19
+ - Macro F1: 0.8920
20
+
21
+ ## Model description
22
+
23
+ More information needed
24
+
25
+ ## Intended uses & limitations
26
+
27
+ More information needed
28
+
29
+ ## Training and evaluation data
30
+
31
+ More information needed
32
+
33
+ ## Training procedure
34
+
35
+ ### Training hyperparameters
36
+
37
+ The following hyperparameters were used during training:
38
+ - learning_rate: 1e-05
39
+ - train_batch_size: 16
40
+ - eval_batch_size: 16
41
+ - seed: 42
42
+ - optimizer: Use OptimizerNames.ADAMW_TORCH_FUSED with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
43
+ - lr_scheduler_type: linear
44
+ - num_epochs: 5
45
+
46
+ ### Training results
47
+
48
+ | Training Loss | Epoch | Step | Validation Loss | Macro F1 |
49
+ |:-------------:|:-----:|:----:|:---------------:|:--------:|
50
+ | 0.9836 | 1.0 | 87 | 0.5345 | 0.7540 |
51
+ | 0.3982 | 2.0 | 174 | 0.3769 | 0.8723 |
52
+ | 0.2025 | 3.0 | 261 | 0.3329 | 0.8814 |
53
+ | 0.1355 | 4.0 | 348 | 0.3793 | 0.8971 |
54
+ | 0.0506 | 5.0 | 435 | 0.4025 | 0.8920 |
55
+
56
+
57
+ ### Framework versions
58
+
59
+ - Transformers 5.0.0
60
+ - Pytorch 2.9.0+cu128
61
+ - Datasets 4.0.0
62
+ - Tokenizers 0.22.2
config.json ADDED
@@ -0,0 +1,47 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "add_cross_attention": false,
3
+ "architectures": [
4
+ "BertForSequenceClassification"
5
+ ],
6
+ "attention_probs_dropout_prob": 0.1,
7
+ "bos_token_id": null,
8
+ "classifier_dropout": null,
9
+ "directionality": "bidi",
10
+ "dtype": "float32",
11
+ "eos_token_id": null,
12
+ "gradient_checkpointing": false,
13
+ "hidden_act": "gelu",
14
+ "hidden_dropout_prob": 0.1,
15
+ "hidden_size": 768,
16
+ "id2label": {
17
+ "0": "negative",
18
+ "1": "neutral",
19
+ "2": "positive"
20
+ },
21
+ "initializer_range": 0.02,
22
+ "intermediate_size": 3072,
23
+ "is_decoder": false,
24
+ "label2id": {
25
+ "negative": 0,
26
+ "neutral": 1,
27
+ "positive": 2
28
+ },
29
+ "layer_norm_eps": 1e-12,
30
+ "max_position_embeddings": 512,
31
+ "model_type": "bert",
32
+ "num_attention_heads": 12,
33
+ "num_hidden_layers": 12,
34
+ "pad_token_id": 0,
35
+ "pooler_fc_size": 768,
36
+ "pooler_num_attention_heads": 12,
37
+ "pooler_num_fc_layers": 3,
38
+ "pooler_size_per_head": 128,
39
+ "pooler_type": "first_token_transform",
40
+ "position_embedding_type": "absolute",
41
+ "problem_type": "single_label_classification",
42
+ "tie_word_embeddings": true,
43
+ "transformers_version": "5.0.0",
44
+ "type_vocab_size": 2,
45
+ "use_cache": false,
46
+ "vocab_size": 100000
47
+ }
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:acf2ca798af5c1a447dc304e41dc07891b57e4c5d2c16347fd7caee872f367e8
3
+ size 651398124
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:91436a227e705dde77bd341d9fdf559d3c40210cadc424d646101999f4b53170
3
+ size 5137