dabbitz's picture
Uploaded new symptom classifier with better and adaptive threshold values, trained with 24K rows of data
438f745 verified
{
"activation": "gelu",
"architectures": [
"DistilBertForSequenceClassification"
],
"attention_dropout": 0.1,
"bos_token_id": null,
"dim": 768,
"dropout": 0.1,
"dtype": "float32",
"eos_token_id": null,
"hidden_dim": 3072,
"id2label": {
"0": "headache",
"1": "dizziness",
"2": "fatigue",
"3": "blurred_vision",
"4": "runny_nose",
"5": "blocked_nose",
"6": "itchy_nose",
"7": "shortness_of_breath",
"8": "coughing",
"9": "sneezing",
"10": "pressure_in_ears",
"11": "pressure_in_face",
"12": "low_fever",
"13": "high_fever",
"14": "diarrhea",
"15": "nausea",
"16": "vomiting",
"17": "stomach_cramps",
"18": "stomach_pain",
"19": "occasional_muscle_ache",
"20": "sore_throat",
"21": "chest_tightness",
"22": "itchy_eyes",
"23": "watery_eyes",
"24": "rashes",
"25": "swelling_of_skin",
"26": "redness_of_skin",
"27": "throat_pain",
"28": "itchy_skin",
"29": "dry_skin",
"30": "blisters",
"31": "dandruff",
"32": "thickened_skin",
"33": "raised_bumps_on_skin",
"34": "blotchy_skin",
"35": "redness_in_eyes",
"36": "sensitivity_to_light",
"37": "swelling_eyes",
"38": "double_vision",
"39": "seeing_faded_colors",
"40": "difficulty_seeing_in_the_dark",
"41": "constant_weight_loss",
"42": "pain_or_numbness_at_hands_and_feet",
"43": "frequent_urination",
"44": "slow_healing_of_cuts_and_bruises",
"45": "constant_hunger",
"46": "burning_skin",
"47": "constipation",
"48": "low_back_pain",
"49": "ear_pain",
"50": "hearing_loss",
"51": "melancholy",
"52": "loss_of_appetite",
"53": "lack_of_sleep",
"54": "palpitation",
"55": "pale_skin",
"56": "pain_during_urination",
"57": "groin_pain",
"58": "no_symptom"
},
"initializer_range": 0.02,
"label2id": {
"blisters": 30,
"blocked_nose": 5,
"blotchy_skin": 34,
"blurred_vision": 3,
"burning_skin": 46,
"chest_tightness": 21,
"constant_hunger": 45,
"constant_weight_loss": 41,
"constipation": 47,
"coughing": 8,
"dandruff": 31,
"diarrhea": 14,
"difficulty_seeing_in_the_dark": 40,
"dizziness": 1,
"double_vision": 38,
"dry_skin": 29,
"ear_pain": 49,
"fatigue": 2,
"frequent_urination": 43,
"groin_pain": 57,
"headache": 0,
"hearing_loss": 50,
"high_fever": 13,
"itchy_eyes": 22,
"itchy_nose": 6,
"itchy_skin": 28,
"lack_of_sleep": 53,
"loss_of_appetite": 52,
"low_back_pain": 48,
"low_fever": 12,
"melancholy": 51,
"nausea": 15,
"no_symptom": 58,
"occasional_muscle_ache": 19,
"pain_during_urination": 56,
"pain_or_numbness_at_hands_and_feet": 42,
"pale_skin": 55,
"palpitation": 54,
"pressure_in_ears": 10,
"pressure_in_face": 11,
"raised_bumps_on_skin": 33,
"rashes": 24,
"redness_in_eyes": 35,
"redness_of_skin": 26,
"runny_nose": 4,
"seeing_faded_colors": 39,
"sensitivity_to_light": 36,
"shortness_of_breath": 7,
"slow_healing_of_cuts_and_bruises": 44,
"sneezing": 9,
"sore_throat": 20,
"stomach_cramps": 17,
"stomach_pain": 18,
"swelling_eyes": 37,
"swelling_of_skin": 25,
"thickened_skin": 32,
"throat_pain": 27,
"vomiting": 16,
"watery_eyes": 23
},
"max_position_embeddings": 512,
"model_type": "distilbert",
"n_heads": 12,
"n_layers": 6,
"pad_token_id": 0,
"problem_type": "multi_label_classification",
"qa_dropout": 0.1,
"seq_classif_dropout": 0.2,
"sinusoidal_pos_embds": false,
"tie_weights_": true,
"tie_word_embeddings": true,
"transformers_version": "5.0.0",
"use_cache": false,
"vocab_size": 30522
}