Hugging Face's logo

Iloriayomide
/
my-symptom-checker-biobert

Text Classification
Transformers
Joblib
Safetensors
bert
Generated from Trainer
text-embeddings-inference
Model card Files
xet
 Community
1
Iloriayomide commited on
Commit
b76c0e9
·
verified ·
1 Parent(s): 38f6cd4

Training in progress, epoch 1

Browse files
Files changed (3) hide show
  1. config.json +48 -48
  2. model.safetensors +1 -1
  3. training_args.bin +1 -1
config.json CHANGED
@@ -9,58 +9,58 @@
9
  "hidden_dropout_prob": 0.1,
10
  "hidden_size": 768,
11
  "id2label": {
12
- "0": 0,
13
- "1": 1,
14
- "2": 2,
15
- "3": 3,
16
- "4": 4,
17
- "5": 5,
18
- "6": 6,
19
- "7": 7,
20
- "8": 8,
21
- "9": 9,
22
- "10": 10,
23
- "11": 11,
24
- "12": 12,
25
- "13": 13,
26
- "14": 14,
27
- "15": 15,
28
- "16": 16,
29
- "17": 17,
30
- "18": 18,
31
- "19": 19,
32
- "20": 20,
33
- "21": 21,
34
- "22": 22,
35
- "23": 23
36
  },
37
  "initializer_range": 0.02,
38
  "intermediate_size": 3072,
39
  "label2id": {
40
- "0": 0,
41
- "1": 1,
42
- "2": 2,
43
- "3": 3,
44
- "4": 4,
45
- "5": 5,
46
- "6": 6,
47
- "7": 7,
48
- "8": 8,
49
- "9": 9,
50
- "10": 10,
51
- "11": 11,
52
- "12": 12,
53
- "13": 13,
54
- "14": 14,
55
- "15": 15,
56
- "16": 16,
57
- "17": 17,
58
- "18": 18,
59
- "19": 19,
60
- "20": 20,
61
- "21": 21,
62
- "22": 22,
63
- "23": 23
64
  },
65
  "layer_norm_eps": 1e-12,
66
  "max_position_embeddings": 512,
 
9
  "hidden_dropout_prob": 0.1,
10
  "hidden_size": 768,
11
  "id2label": {
12
+ "0": "Acne",
13
+ "1": "Arthritis",
14
+ "2": "Bronchial Asthma",
15
+ "3": "Cervical spondylosis",
16
+ "4": "Chicken pox",
17
+ "5": "Common Cold",
18
+ "6": "Dengue",
19
+ "7": "Dimorphic Hemorrhoids",
20
+ "8": "Fungal infection",
21
+ "9": "Hypertension",
22
+ "10": "Impetigo",
23
+ "11": "Jaundice",
24
+ "12": "Malaria",
25
+ "13": "Migraine",
26
+ "14": "Pneumonia",
27
+ "15": "Psoriasis",
28
+ "16": "Typhoid",
29
+ "17": "Varicose Veins",
30
+ "18": "allergy",
31
+ "19": "diabetes",
32
+ "20": "drug reaction",
33
+ "21": "gastroesophageal reflux disease",
34
+ "22": "peptic ulcer disease",
35
+ "23": "urinary tract infection"
36
  },
37
  "initializer_range": 0.02,
38
  "intermediate_size": 3072,
39
  "label2id": {
40
+ "Acne": 0,
41
+ "Arthritis": 1,
42
+ "Bronchial Asthma": 2,
43
+ "Cervical spondylosis": 3,
44
+ "Chicken pox": 4,
45
+ "Common Cold": 5,
46
+ "Dengue": 6,
47
+ "Dimorphic Hemorrhoids": 7,
48
+ "Fungal infection": 8,
49
+ "Hypertension": 9,
50
+ "Impetigo": 10,
51
+ "Jaundice": 11,
52
+ "Malaria": 12,
53
+ "Migraine": 13,
54
+ "Pneumonia": 14,
55
+ "Psoriasis": 15,
56
+ "Typhoid": 16,
57
+ "Varicose Veins": 17,
58
+ "allergy": 18,
59
+ "diabetes": 19,
60
+ "drug reaction": 20,
61
+ "gastroesophageal reflux disease": 21,
62
+ "peptic ulcer disease": 22,
63
+ "urinary tract infection": 23
64
  },
65
  "layer_norm_eps": 1e-12,
66
  "max_position_embeddings": 512,
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:2fc0ab401f668bd53a3a9cca63efc2df047541d1fc5edd804e957fe7c242cc2a
3
  size 433338440
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a0e250983a8ba42fb34e94208c4ff11484e948a697d5085db3994666ed586ae0
3
  size 433338440
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:6f84823444096676d080eda0c802fa3c990f98fba4734d207d2258d2180e1452
3
  size 5841
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e9ca1bbbf336ae7bb0840763e20c41ffe7f2c4e825c733ae4c99b135cf80592c
3
  size 5841