{ "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 }