Upload TFDistilBertForTokenClassification

README.md

@@ -15,8 +15,8 @@ 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.1913
-- Validation Loss: 0.3219
+- Train Loss: 0.1894
+- Validation Loss: 0.3447
 - Epoch: 2
 
 ## Model description
@@ -43,9 +43,9 @@ The following hyperparameters were used during training:
 
 | Train Loss | Validation Loss | Epoch |
 |:----------:|:---------------:|:-----:|
-| 0.5249     | 0.3549          | 0     |
-| 0.2824     | 0.3316          | 1     |
-| 0.1913     | 0.3219          | 2     |
+| 0.5347     | 0.3843          | 0     |
+| 0.2871     | 0.3327          | 1     |
+| 0.1894     | 0.3447          | 2     |
 
 
 ### Framework versions

config.json

@@ -12,23 +12,23 @@
     "0": "O",
     "1": "B-allergy_name",
     "2": "I-allergy_name",
-    "3": "B-cancer",
-    "4": "I-cancer",
-    "5": "B-chronic_disease",
-    "6": "I-chronic_disease",
-    "7": "B-treatment",
-    "8": "I-treatment"
+    "3": "B-treatment",
+    "4": "I-treatment",
+    "5": "B-cancer",
+    "6": "I-cancer",
+    "7": "B-chronic_disease",
+    "8": "I-chronic_disease"
   },
   "initializer_range": 0.02,
   "label2id": {
     "B-allergy_name": 1,
-    "B-cancer": 3,
-    "B-chronic_disease": 5,
-    "B-treatment": 7,
+    "B-cancer": 5,
+    "B-chronic_disease": 7,
+    "B-treatment": 3,
     "I-allergy_name": 2,
-    "I-cancer": 4,
-    "I-chronic_disease": 6,
-    "I-treatment": 8,
+    "I-cancer": 6,
+    "I-chronic_disease": 8,
+    "I-treatment": 4,
     "O": 0
   },
   "max_position_embeddings": 512,

tf_model.h5

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:93346d02c5635935e1fb66b6b7c7256eb0cdd95fb1a0ab30bf404aa7cdc105bf
+oid sha256:dbb07f2fe43ab67b8e68ed07125a62b0a7f596ca4f9e1bf08d3c8a53562fc42f
 size 265606416