Commit
•
f7ed048
1
Parent(s):
d7a4c22
Update config.json
Browse files- config.json +11 -13
config.json
CHANGED
|
@@ -1,9 +1,7 @@
|
|
| 1 |
{
|
| 2 |
"_name_or_path": "distilbert-base-uncased",
|
| 3 |
"activation": "gelu",
|
| 4 |
-
"architectures": [
|
| 5 |
-
"DistilBertForSequenceClassification"
|
| 6 |
-
],
|
| 7 |
"attention_dropout": 0.1,
|
| 8 |
"dim": 768,
|
| 9 |
"dropout": 0.1,
|
|
@@ -39,6 +37,14 @@
|
|
| 39 |
"label2id": {
|
| 40 |
"emotional pain": 0,
|
| 41 |
"hair falling out": 1,
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 42 |
"joint pain": 10,
|
| 43 |
"hard to breath": 11,
|
| 44 |
"head ache": 12,
|
|
@@ -49,19 +55,11 @@
|
|
| 49 |
"internal pain": 17,
|
| 50 |
"blurry vision": 18,
|
| 51 |
"acne": 19,
|
| 52 |
-
"LABEL_2": 2,
|
| 53 |
"muscle pain": 20,
|
| 54 |
"neck pain": 21,
|
| 55 |
"cough": 22,
|
| 56 |
"ear ache": 23,
|
| 57 |
-
"feeling cold": 24
|
| 58 |
-
"infected wound": 3,
|
| 59 |
-
"foot ache": 4,
|
| 60 |
-
"shoulder pain": 5,
|
| 61 |
-
"injury from sports": 6,
|
| 62 |
-
"skin issue": 7,
|
| 63 |
-
"stomach ache": 8,
|
| 64 |
-
"knee pain": 9
|
| 65 |
},
|
| 66 |
"max_position_embeddings": 512,
|
| 67 |
"model_type": "distilbert",
|
|
@@ -76,4 +74,4 @@
|
|
| 76 |
"torch_dtype": "float32",
|
| 77 |
"transformers_version": "4.30.2",
|
| 78 |
"vocab_size": 30522
|
| 79 |
-
}
|
|
|
|
| 1 |
{
|
| 2 |
"_name_or_path": "distilbert-base-uncased",
|
| 3 |
"activation": "gelu",
|
| 4 |
+
"architectures": ["DistilBertForSequenceClassification"],
|
|
|
|
|
|
|
| 5 |
"attention_dropout": 0.1,
|
| 6 |
"dim": 768,
|
| 7 |
"dropout": 0.1,
|
|
|
|
| 37 |
"label2id": {
|
| 38 |
"emotional pain": 0,
|
| 39 |
"hair falling out": 1,
|
| 40 |
+
"heart hurts": 2,
|
| 41 |
+
"infected wound": 3,
|
| 42 |
+
"foot ache": 4,
|
| 43 |
+
"shoulder pain": 5,
|
| 44 |
+
"injury from sports": 6,
|
| 45 |
+
"skin issue": 7,
|
| 46 |
+
"stomach ache": 8,
|
| 47 |
+
"knee pain": 9,
|
| 48 |
"joint pain": 10,
|
| 49 |
"hard to breath": 11,
|
| 50 |
"head ache": 12,
|
|
|
|
| 55 |
"internal pain": 17,
|
| 56 |
"blurry vision": 18,
|
| 57 |
"acne": 19,
|
|
|
|
| 58 |
"muscle pain": 20,
|
| 59 |
"neck pain": 21,
|
| 60 |
"cough": 22,
|
| 61 |
"ear ache": 23,
|
| 62 |
+
"feeling cold": 24
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 63 |
},
|
| 64 |
"max_position_embeddings": 512,
|
| 65 |
"model_type": "distilbert",
|
|
|
|
| 74 |
"torch_dtype": "float32",
|
| 75 |
"transformers_version": "4.30.2",
|
| 76 |
"vocab_size": 30522
|
| 77 |
+
}
|