Symptom_Prediction / config.json
Iloriayomide's picture
End of training
bee9179 verified
{
"architectures": [
"BertForSequenceClassification"
],
"attention_probs_dropout_prob": 0.1,
"classifier_dropout": null,
"hidden_act": "gelu",
"hidden_dropout_prob": 0.1,
"hidden_size": 768,
"id2label": {
"0": "Acne",
"1": "Arthritis",
"2": "Bronchial Asthma",
"3": "Cervical spondylosis",
"4": "Chicken pox",
"5": "Common Cold",
"6": "Dengue",
"7": "Dimorphic Hemorrhoids",
"8": "Fungal infection",
"9": "Hypertension",
"10": "Impetigo",
"11": "Jaundice",
"12": "Malaria",
"13": "Migraine",
"14": "Pneumonia",
"15": "Psoriasis",
"16": "Typhoid",
"17": "Varicose Veins",
"18": "actinic keratosis",
"19": "acute bronchiolitis",
"20": "acute bronchitis",
"21": "acute bronchospasm",
"22": "acute kidney injury",
"23": "acute pancreatitis",
"24": "acute sinusitis",
"25": "allergy",
"26": "angina",
"27": "anxiety",
"28": "appendicitis",
"29": "arthritis of the hip",
"30": "asthma",
"31": "benign prostatic hyperplasia (bph)",
"32": "brachial neuritis",
"33": "bursitis",
"34": "carpal tunnel syndrome",
"35": "cholecystitis",
"36": "chronic back pain",
"37": "chronic constipation",
"38": "chronic obstructive pulmonary disease (copd)",
"39": "common cold",
"40": "complex regional pain syndrome",
"41": "concussion",
"42": "conjunctivitis",
"43": "conjunctivitis due to allergy",
"44": "contact dermatitis",
"45": "cornea infection",
"46": "croup",
"47": "cystitis",
"48": "degenerative disc disease",
"49": "dental caries",
"50": "depression",
"51": "developmental disability",
"52": "diabetes",
"53": "diaper rash",
"54": "diverticulitis",
"55": "drug reaction",
"56": "ear drum damage",
"57": "eczema",
"58": "esophagitis",
"59": "eustachian tube dysfunction (ear disorder)",
"60": "fungal infection of the hair",
"61": "gallstone",
"62": "gastroesophageal reflux disease",
"63": "gastrointestinal hemorrhage",
"64": "gout",
"65": "gum disease",
"66": "heart attack",
"67": "heart failure",
"68": "hemorrhoids",
"69": "herniated disk",
"70": "hiatal hernia",
"71": "hyperemesis gravidarum",
"72": "hypertensive heart disease",
"73": "hypoglycemia",
"74": "idiopathic excessive menstruation",
"75": "idiopathic irregular menstrual cycle",
"76": "idiopathic painful menstruation",
"77": "infectious gastroenteritis",
"78": "injury to the arm",
"79": "injury to the leg",
"80": "injury to the trunk",
"81": "liver disease",
"82": "macular degeneration",
"83": "marijuana abuse",
"84": "multiple sclerosis",
"85": "noninfectious gastroenteritis",
"86": "nose disorder",
"87": "obstructive sleep apnea (osa)",
"88": "otitis externa (swimmer's ear)",
"89": "otitis media",
"90": "pain after an operation",
"91": "panic disorder",
"92": "pelvic inflammatory disease",
"93": "peptic ulcer disease",
"94": "peripheral nerve disorder",
"95": "personality disorder",
"96": "pneumonia",
"97": "problem during pregnancy",
"98": "psoriasis",
"99": "pyogenic skin infection",
"100": "rectal disorder",
"101": "schizophrenia",
"102": "seasonal allergies (hay fever)",
"103": "sebaceous cyst",
"104": "sepsis",
"105": "sickle cell crisis",
"106": "sinus bradycardia",
"107": "skin pigmentation disorder",
"108": "skin polyp",
"109": "spinal stenosis",
"110": "spondylosis",
"111": "spontaneous abortion",
"112": "sprain or strain",
"113": "strep throat",
"114": "stye",
"115": "temporary or benign blood in urine",
"116": "threatened pregnancy",
"117": "urinary tract infection",
"118": "vaginal cyst",
"119": "vaginitis",
"120": "vulvodynia"
},
"initializer_range": 0.02,
"intermediate_size": 3072,
"label2id": {
"Acne": 0,
"Arthritis": 1,
"Bronchial Asthma": 2,
"Cervical spondylosis": 3,
"Chicken pox": 4,
"Common Cold": 5,
"Dengue": 6,
"Dimorphic Hemorrhoids": 7,
"Fungal infection": 8,
"Hypertension": 9,
"Impetigo": 10,
"Jaundice": 11,
"Malaria": 12,
"Migraine": 13,
"Pneumonia": 14,
"Psoriasis": 15,
"Typhoid": 16,
"Varicose Veins": 17,
"actinic keratosis": 18,
"acute bronchiolitis": 19,
"acute bronchitis": 20,
"acute bronchospasm": 21,
"acute kidney injury": 22,
"acute pancreatitis": 23,
"acute sinusitis": 24,
"allergy": 25,
"angina": 26,
"anxiety": 27,
"appendicitis": 28,
"arthritis of the hip": 29,
"asthma": 30,
"benign prostatic hyperplasia (bph)": 31,
"brachial neuritis": 32,
"bursitis": 33,
"carpal tunnel syndrome": 34,
"cholecystitis": 35,
"chronic back pain": 36,
"chronic constipation": 37,
"chronic obstructive pulmonary disease (copd)": 38,
"common cold": 39,
"complex regional pain syndrome": 40,
"concussion": 41,
"conjunctivitis": 42,
"conjunctivitis due to allergy": 43,
"contact dermatitis": 44,
"cornea infection": 45,
"croup": 46,
"cystitis": 47,
"degenerative disc disease": 48,
"dental caries": 49,
"depression": 50,
"developmental disability": 51,
"diabetes": 52,
"diaper rash": 53,
"diverticulitis": 54,
"drug reaction": 55,
"ear drum damage": 56,
"eczema": 57,
"esophagitis": 58,
"eustachian tube dysfunction (ear disorder)": 59,
"fungal infection of the hair": 60,
"gallstone": 61,
"gastroesophageal reflux disease": 62,
"gastrointestinal hemorrhage": 63,
"gout": 64,
"gum disease": 65,
"heart attack": 66,
"heart failure": 67,
"hemorrhoids": 68,
"herniated disk": 69,
"hiatal hernia": 70,
"hyperemesis gravidarum": 71,
"hypertensive heart disease": 72,
"hypoglycemia": 73,
"idiopathic excessive menstruation": 74,
"idiopathic irregular menstrual cycle": 75,
"idiopathic painful menstruation": 76,
"infectious gastroenteritis": 77,
"injury to the arm": 78,
"injury to the leg": 79,
"injury to the trunk": 80,
"liver disease": 81,
"macular degeneration": 82,
"marijuana abuse": 83,
"multiple sclerosis": 84,
"noninfectious gastroenteritis": 85,
"nose disorder": 86,
"obstructive sleep apnea (osa)": 87,
"otitis externa (swimmer's ear)": 88,
"otitis media": 89,
"pain after an operation": 90,
"panic disorder": 91,
"pelvic inflammatory disease": 92,
"peptic ulcer disease": 93,
"peripheral nerve disorder": 94,
"personality disorder": 95,
"pneumonia": 96,
"problem during pregnancy": 97,
"psoriasis": 98,
"pyogenic skin infection": 99,
"rectal disorder": 100,
"schizophrenia": 101,
"seasonal allergies (hay fever)": 102,
"sebaceous cyst": 103,
"sepsis": 104,
"sickle cell crisis": 105,
"sinus bradycardia": 106,
"skin pigmentation disorder": 107,
"skin polyp": 108,
"spinal stenosis": 109,
"spondylosis": 110,
"spontaneous abortion": 111,
"sprain or strain": 112,
"strep throat": 113,
"stye": 114,
"temporary or benign blood in urine": 115,
"threatened pregnancy": 116,
"urinary tract infection": 117,
"vaginal cyst": 118,
"vaginitis": 119,
"vulvodynia": 120
},
"layer_norm_eps": 1e-12,
"max_position_embeddings": 512,
"model_type": "bert",
"num_attention_heads": 12,
"num_hidden_layers": 12,
"pad_token_id": 0,
"position_embedding_type": "absolute",
"problem_type": "single_label_classification",
"torch_dtype": "float32",
"transformers_version": "4.53.3",
"type_vocab_size": 2,
"use_cache": true,
"vocab_size": 28996
}