Added more classes by cleaning data
Browse files- config.json +43 -13
- tf_model.h5 +2 -2
config.json
CHANGED
@@ -9,20 +9,50 @@
|
|
9 |
"dropout": 0.1,
|
10 |
"hidden_dim": 3072,
|
11 |
"id2label": {
|
12 |
-
"0": "
|
13 |
-
"1": "
|
14 |
-
"2": "
|
15 |
-
"3": "
|
16 |
-
"4": "
|
17 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
18 |
"initializer_range": 0.02,
|
19 |
"label2id": {
|
20 |
-
"
|
21 |
-
"
|
22 |
-
"
|
23 |
-
"
|
24 |
-
"
|
25 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
26 |
"max_position_embeddings": 512,
|
27 |
"model_type": "distilbert",
|
28 |
"n_heads": 12,
|
@@ -32,6 +62,6 @@
|
|
32 |
"seq_classif_dropout": 0.2,
|
33 |
"sinusoidal_pos_embds": false,
|
34 |
"tie_weights_": true,
|
35 |
-
"transformers_version": "4.29.
|
36 |
"vocab_size": 30522
|
37 |
}
|
|
|
9 |
"dropout": 0.1,
|
10 |
"hidden_dim": 3072,
|
11 |
"id2label": {
|
12 |
+
"0": "Cardiology",
|
13 |
+
"1": "Dentistry",
|
14 |
+
"2": "Dermatology",
|
15 |
+
"3": "Diets and nutritions",
|
16 |
+
"4": "Endocrinology",
|
17 |
+
"5": "Ent",
|
18 |
+
"6": "Gastroenterology",
|
19 |
+
"7": "Gynecology",
|
20 |
+
"8": "Hematology - oncology",
|
21 |
+
"9": "Hospice care",
|
22 |
+
"10": "Nephrology",
|
23 |
+
"11": "Neurology",
|
24 |
+
"12": "Opd",
|
25 |
+
"13": "Ophthalmology",
|
26 |
+
"14": "Orthopedic",
|
27 |
+
"15": "Orthopedists",
|
28 |
+
"16": "Pediatrics",
|
29 |
+
"17": "Psychiatry",
|
30 |
+
"18": "Radiology",
|
31 |
+
"19": "Urology"
|
32 |
+
},
|
33 |
"initializer_range": 0.02,
|
34 |
"label2id": {
|
35 |
+
"Cardiology": 0,
|
36 |
+
"Dentistry": 1,
|
37 |
+
"Dermatology": 2,
|
38 |
+
"Diets and nutritions": 3,
|
39 |
+
"Endocrinology": 4,
|
40 |
+
"Ent": 5,
|
41 |
+
"Gastroenterology": 6,
|
42 |
+
"Gynecology": 7,
|
43 |
+
"Hematology - oncology": 8,
|
44 |
+
"Hospice care": 9,
|
45 |
+
"Nephrology": 10,
|
46 |
+
"Neurology": 11,
|
47 |
+
"Opd": 12,
|
48 |
+
"Ophthalmology": 13,
|
49 |
+
"Orthopedic": 14,
|
50 |
+
"Orthopedists": 15,
|
51 |
+
"Pediatrics": 16,
|
52 |
+
"Psychiatry": 17,
|
53 |
+
"Radiology": 18,
|
54 |
+
"Urology": 19
|
55 |
+
},
|
56 |
"max_position_embeddings": 512,
|
57 |
"model_type": "distilbert",
|
58 |
"n_heads": 12,
|
|
|
62 |
"seq_classif_dropout": 0.2,
|
63 |
"sinusoidal_pos_embds": false,
|
64 |
"tie_weights_": true,
|
65 |
+
"transformers_version": "4.29.2",
|
66 |
"vocab_size": 30522
|
67 |
}
|
tf_model.h5
CHANGED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
-
size
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:55e0b05b3eb4fa30e3c1d3e809b0a8c69d2e2fdfba50b0467e0d60daf4b25fa2
|
3 |
+
size 268010440
|