Added more classes by cleaning data

config.json

@@ -9,20 +9,50 @@
   "dropout": 0.1,
   "hidden_dim": 3072,
   "id2label": {
-    "0": "LABEL_0",
-    "1": "LABEL_1",
-    "2": "LABEL_2",
-    "3": "LABEL_3",
-    "4": "LABEL_4"
-  },
+    "0": "Cardiology",
+    "1": "Dentistry",
+    "2": "Dermatology",
+    "3": "Diets and nutritions",
+    "4": "Endocrinology",
+    "5": "Ent",
+    "6": "Gastroenterology",
+    "7": "Gynecology",
+    "8": "Hematology - oncology",
+    "9": "Hospice care",
+    "10": "Nephrology",
+    "11": "Neurology",
+    "12": "Opd",
+    "13": "Ophthalmology",
+    "14": "Orthopedic",
+    "15": "Orthopedists",
+    "16": "Pediatrics",
+    "17": "Psychiatry",
+    "18": "Radiology",
+    "19": "Urology"
+},
   "initializer_range": 0.02,
   "label2id": {
-    "LABEL_0": 0,
-    "LABEL_1": 1,
-    "LABEL_2": 2,
-    "LABEL_3": 3,
-    "LABEL_4": 4
-  },
+    "Cardiology": 0,
+    "Dentistry": 1,
+    "Dermatology": 2,
+    "Diets and nutritions": 3,
+    "Endocrinology": 4,
+    "Ent": 5,
+    "Gastroenterology": 6,
+    "Gynecology": 7,
+    "Hematology - oncology": 8,
+    "Hospice care": 9,
+    "Nephrology": 10,
+    "Neurology": 11,
+    "Opd": 12,
+    "Ophthalmology": 13,
+    "Orthopedic": 14,
+    "Orthopedists": 15,
+    "Pediatrics": 16,
+    "Psychiatry": 17,
+    "Radiology": 18,
+    "Urology": 19
+},
   "max_position_embeddings": 512,
   "model_type": "distilbert",
   "n_heads": 12,
@@ -32,6 +62,6 @@
   "seq_classif_dropout": 0.2,
   "sinusoidal_pos_embds": false,
   "tie_weights_": true,
-  "transformers_version": "4.29.1",
+  "transformers_version": "4.29.2",
   "vocab_size": 30522
 }

tf_model.h5

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:da574063e92fe095da263c692b1b4c6f9ba283fef8478fba225390b1a6a0dc4b
-size 267964360
+oid sha256:55e0b05b3eb4fa30e3c1d3e809b0a8c69d2e2fdfba50b0467e0d60daf4b25fa2
+size 268010440