Training complete

README.md

@@ -3,11 +3,26 @@ license: mit
 base_model: xlm-roberta-base
 tags:
 - generated_from_trainer
+datasets:
+- masakhaner2
 metrics:
 - f1
 model-index:
 - name: xlm-roberta-base-finetuned-wolof
-  results: []
+  results:
+  - task:
+      name: Token Classification
+      type: token-classification
+    dataset:
+      name: masakhaner2
+      type: masakhaner2
+      config: wol
+      split: validation
+      args: wol
+    metrics:
+    - name: F1
+      type: f1
+      value: 0.8217446270543616
 ---
 
 <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -15,10 +30,10 @@ should probably proofread and complete it, then remove this comment. -->
 
 # xlm-roberta-base-finetuned-wolof
 
-This model is a fine-tuned version of [xlm-roberta-base](https://huggingface.co/xlm-roberta-base) on the None dataset.
+This model is a fine-tuned version of [xlm-roberta-base](https://huggingface.co/xlm-roberta-base) on the masakhaner2 dataset.
 It achieves the following results on the evaluation set:
-- Loss: 0.0600
-- F1: 0.7306
+- Loss: 0.0589
+- F1: 0.8217
 
 ## Model description
 
@@ -49,9 +64,9 @@ The following hyperparameters were used during training:
 
 | Training Loss | Epoch | Step | Validation Loss | F1     |
 |:-------------:|:-----:|:----:|:---------------:|:------:|
-| 0.3197        | 1.0   | 76   | 0.0961          | 0.4105 |
-| 0.1061        | 2.0   | 152  | 0.0692          | 0.6769 |
-| 0.061         | 3.0   | 228  | 0.0600          | 0.7306 |
+| 0.2113        | 1.0   | 247  | 0.0823          | 0.7443 |
+| 0.0749        | 2.0   | 494  | 0.0635          | 0.8097 |
+| 0.0445        | 3.0   | 741  | 0.0589          | 0.8217 |
 
 
 ### Framework versions

config.json

@@ -12,26 +12,26 @@
   "hidden_size": 768,
   "id2label": {
     "0": "O",
-    "1": "B-LOC",
-    "2": "B-PER",
-    "3": "I-PER",
-    "4": "B-ORG",
-    "5": "I-DATE",
-    "6": "B-DATE",
-    "7": "I-ORG",
-    "8": "I-LOC"
+    "1": "B-PER",
+    "2": "I-PER",
+    "3": "B-ORG",
+    "4": "I-ORG",
+    "5": "B-LOC",
+    "6": "I-LOC",
+    "7": "B-DATE",
+    "8": "I-DATE"
   },
   "initializer_range": 0.02,
   "intermediate_size": 3072,
   "label2id": {
-    "B-DATE": 6,
-    "B-LOC": 1,
-    "B-ORG": 4,
-    "B-PER": 2,
-    "I-DATE": 5,
-    "I-LOC": 8,
-    "I-ORG": 7,
-    "I-PER": 3,
+    "B-DATE": 7,
+    "B-LOC": 5,
+    "B-ORG": 3,
+    "B-PER": 1,
+    "I-DATE": 8,
+    "I-LOC": 6,
+    "I-ORG": 4,
+    "I-PER": 2,
     "O": 0
   },
   "layer_norm_eps": 1e-05,

pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:74f8233ab0f6fba915fdbd840015bf7ebec2efc0b9be1a27783eb715854120cf
+oid sha256:4cf9949a7ea4a781f235c600cf096450ee7d2643360a168c35a7740261193d01
 size 1109908201

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:69430cae915e86ffb5d3c44992ed7951328a294449f28c07dad4bb39191afbed
+oid sha256:668a15c50a1956199f0b05400b7aae86837591a5e8b69af8ed8a7e46c7015f9f
 size 4091