Training in progress epoch 0

README.md

@@ -14,13 +14,13 @@ probably proofread and complete it, then remove this comment. -->
 
 This model is a fine-tuned version of [distilbert-base-uncased](https://huggingface.co/distilbert-base-uncased) on an unknown dataset.
 It achieves the following results on the evaluation set:
-- Train Loss: 0.0282
-- Validation Loss: 0.0510
-- Train Precision: 0.9227
-- Train Recall: 0.9355
-- Train F1: 0.9291
-- Train Accuracy: 0.9864
-- Epoch: 2
+- Train Loss: 0.1611
+- Validation Loss: 0.0606
+- Train Precision: 0.8987
+- Train Recall: 0.9164
+- Train F1: 0.9074
+- Train Accuracy: 0.9830
+- Epoch: 0
 
 ## Model description
 
@@ -46,9 +46,7 @@ The following hyperparameters were used during training:
 
 | Train Loss | Validation Loss | Train Precision | Train Recall | Train F1 | Train Accuracy | Epoch |
 |:----------:|:---------------:|:---------------:|:------------:|:--------:|:--------------:|:-----:|
-| 0.1833     | 0.0574          | 0.9038          | 0.9217       | 0.9127   | 0.9838         | 0     |
-| 0.0444     | 0.0510          | 0.9236          | 0.9318       | 0.9277   | 0.9858         | 1     |
-| 0.0282     | 0.0510          | 0.9227          | 0.9355       | 0.9291   | 0.9864         | 2     |
+| 0.1611     | 0.0606          | 0.8987          | 0.9164       | 0.9074   | 0.9830         | 0     |
 
 
 ### Framework versions

config.json

@@ -10,33 +10,25 @@
   "hidden_dim": 3072,
   "id2label": {
     "0": "O",
-    "1": "B-corporation",
-    "2": "I-corporation",
-    "3": "B-creative-work",
-    "4": "I-creative-work",
-    "5": "B-group",
-    "6": "I-group",
-    "7": "B-location",
-    "8": "I-location",
-    "9": "B-person",
-    "10": "I-person",
-    "11": "B-product",
-    "12": "I-product"
+    "1": "B-PER",
+    "2": "I-PER",
+    "3": "B-ORG",
+    "4": "I-ORG",
+    "5": "B-LOC",
+    "6": "I-LOC",
+    "7": "B-MISC",
+    "8": "I-MISC"
   },
   "initializer_range": 0.02,
   "label2id": {
-    "B-corporation": 1,
-    "B-creative-work": 3,
-    "B-group": 5,
-    "B-location": 7,
-    "B-person": 9,
-    "B-product": 11,
-    "I-corporation": 2,
-    "I-creative-work": 4,
-    "I-group": 6,
-    "I-location": 8,
-    "I-person": 10,
-    "I-product": 12,
+    "B-LOC": 5,
+    "B-MISC": 7,
+    "B-ORG": 3,
+    "B-PER": 1,
+    "I-LOC": 6,
+    "I-MISC": 8,
+    "I-ORG": 4,
+    "I-PER": 2,
     "O": 0
   },
   "max_position_embeddings": 512,

tf_model.h5

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:386f6970f77a94db331520fac1f13e8d467b1e835484e4a67a266ed0ce97f7bc
-size 265618792
+oid sha256:8edbe401a548cb7cb2fe1164b5cdc3aaf18eca2f78929afa4e713c717535e4dc
+size 265606504