Beijuka commited on
Commit
bada58f
·
verified ·
1 Parent(s): 92e1065

End of training

Browse files
Files changed (2) hide show
  1. README.md +104 -0
  2. model.safetensors +1 -1
README.md ADDED
@@ -0,0 +1,104 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ library_name: transformers
3
+ license: afl-3.0
4
+ base_model: masakhane/afroxlmr-large-ner-masakhaner-1.0_2.0
5
+ tags:
6
+ - named-entity-recognition
7
+ - lumasaba
8
+ - african-language
9
+ - pii-detection
10
+ - token-classification
11
+ - generated_from_trainer
12
+ datasets:
13
+ - Beijuka/Multilingual_PII_NER_dataset
14
+ metrics:
15
+ - precision
16
+ - recall
17
+ - f1
18
+ - accuracy
19
+ model-index:
20
+ - name: multilingual-masakhane/afroxlmr-large-ner-masakhaner-1.0_2.0-lumasaba-ner-v1
21
+ results:
22
+ - task:
23
+ name: Token Classification
24
+ type: token-classification
25
+ dataset:
26
+ name: Beijuka/Multilingual_PII_NER_dataset
27
+ type: Beijuka/Multilingual_PII_NER_dataset
28
+ args: 'split: train+validation+test'
29
+ metrics:
30
+ - name: Precision
31
+ type: precision
32
+ value: 0.9702892885066459
33
+ - name: Recall
34
+ type: recall
35
+ value: 0.9487767584097859
36
+ - name: F1
37
+ type: f1
38
+ value: 0.9594124468496328
39
+ - name: Accuracy
40
+ type: accuracy
41
+ value: 0.9525409491810164
42
+ ---
43
+
44
+ <!-- This model card has been generated automatically according to the information the Trainer had access to. You
45
+ should probably proofread and complete it, then remove this comment. -->
46
+
47
+ # multilingual-masakhane/afroxlmr-large-ner-masakhaner-1.0_2.0-lumasaba-ner-v1
48
+
49
+ This model is a fine-tuned version of [masakhane/afroxlmr-large-ner-masakhaner-1.0_2.0](https://huggingface.co/masakhane/afroxlmr-large-ner-masakhaner-1.0_2.0) on the Beijuka/Multilingual_PII_NER_dataset dataset.
50
+ It achieves the following results on the evaluation set:
51
+ - Loss: 0.3834
52
+ - Precision: 0.9703
53
+ - Recall: 0.9488
54
+ - F1: 0.9594
55
+ - Accuracy: 0.9525
56
+
57
+ ## Model description
58
+
59
+ More information needed
60
+
61
+ ## Intended uses & limitations
62
+
63
+ More information needed
64
+
65
+ ## Training and evaluation data
66
+
67
+ More information needed
68
+
69
+ ## Training procedure
70
+
71
+ ### Training hyperparameters
72
+
73
+ The following hyperparameters were used during training:
74
+ - learning_rate: 2e-05
75
+ - train_batch_size: 4
76
+ - eval_batch_size: 4
77
+ - seed: 42
78
+ - optimizer: Use OptimizerNames.ADAMW_TORCH_FUSED with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
79
+ - lr_scheduler_type: linear
80
+ - num_epochs: 20
81
+
82
+ ### Training results
83
+
84
+ | Training Loss | Epoch | Step | Validation Loss | Precision | Recall | F1 | Accuracy |
85
+ |:-------------:|:-----:|:----:|:---------------:|:---------:|:------:|:------:|:--------:|
86
+ | 1.1185 | 1.0 | 796 | 0.5047 | 0.8681 | 0.8810 | 0.8745 | 0.8735 |
87
+ | 0.3868 | 2.0 | 1592 | 0.4627 | 0.9012 | 0.9146 | 0.9079 | 0.9108 |
88
+ | 0.2335 | 3.0 | 2388 | 0.4419 | 0.9115 | 0.9272 | 0.9193 | 0.9198 |
89
+ | 0.1462 | 4.0 | 3184 | 0.3402 | 0.9499 | 0.9507 | 0.9503 | 0.9520 |
90
+ | 0.1072 | 5.0 | 3980 | 0.2399 | 0.9560 | 0.9538 | 0.9549 | 0.9563 |
91
+ | 0.0916 | 6.0 | 4776 | 0.3072 | 0.9548 | 0.9593 | 0.9570 | 0.9588 |
92
+ | 0.0432 | 7.0 | 5572 | 0.3124 | 0.9573 | 0.9663 | 0.9618 | 0.9605 |
93
+ | 0.0383 | 8.0 | 6368 | 0.3386 | 0.9669 | 0.9608 | 0.9639 | 0.9575 |
94
+ | 0.0502 | 9.0 | 7164 | 0.4429 | 0.9644 | 0.9554 | 0.9599 | 0.9550 |
95
+ | 0.0349 | 10.0 | 7960 | 0.4191 | 0.9605 | 0.9522 | 0.9564 | 0.9481 |
96
+ | 0.039 | 11.0 | 8756 | 0.4815 | 0.9558 | 0.9648 | 0.9602 | 0.9537 |
97
+
98
+
99
+ ### Framework versions
100
+
101
+ - Transformers 4.55.4
102
+ - Pytorch 2.8.0+cu126
103
+ - Datasets 4.0.0
104
+ - Tokenizers 0.21.4
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:6dd1b1be1ede1f35ac3ae5b653dea3b5c5b32482502a42e42972cdcf7f98ddd5
3
  size 2235530764
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c8f1cdff7c235be9b981677cfd6485c6fbeb9367780eaf580dc4a23430d015a2
3
  size 2235530764