Jacaranda's picture
Upload XLMRobertaForSequenceClassification
9cfff06
{
"_name_or_path": "/content/drive/MyDrive/Jaca80ENS/checkpoint",
"architectures": [
"XLMRobertaForSequenceClassification"
],
"attention_probs_dropout_prob": 0.1,
"bos_token_id": 0,
"classifier_dropout": null,
"eos_token_id": 2,
"hidden_act": "gelu",
"hidden_dropout_prob": 0.1,
"hidden_size": 1024,
"id2label": {
"0": "abortion",
"1": "ambulance",
"2": "anc_visit",
"3": "baby_acne_pimples",
"4": "baby_breathing",
"5": "baby_colic",
"6": "baby_constipation",
"7": "baby_cord",
"8": "baby_diarrhoea",
"9": "baby_fever_discharge",
"10": "baby_general",
"11": "baby_hiccups",
"12": "baby_jaundice",
"13": "baby_milestone_general",
"14": "baby_milestone_sitting",
"15": "baby_milestone_teething",
"16": "baby_poop",
"17": "baby_rash",
"18": "baby_sex_determination",
"19": "baby_spitting_up",
"20": "bleeding",
"21": "blood_group",
"22": "blood_pressure",
"23": "breastfeeding",
"24": "c_section",
"25": "constipation",
"26": "covid-19_vaccine",
"27": "danger_signs",
"28": "delivery_bag_items",
"29": "depression",
"30": "diarrhoea",
"31": "diet_nutrition",
"32": "discharge",
"33": "dizziness",
"34": "edd",
"35": "family_planning",
"36": "fatigue",
"37": "fetal_movement",
"38": "fever",
"39": "financial",
"40": "hb_level",
"41": "heart_rate_shortness_of_breath",
"42": "heartburn",
"43": "hiv",
"44": "ifas",
"45": "intercourse",
"46": "itching_body",
"47": "labour_signs",
"48": "linda_mama",
"49": "medication_general",
"50": "miscarriage_baby_loss",
"51": "mobility_travelling",
"52": "mosquito",
"53": "nausea_vomiting",
"54": "numbness",
"55": "ok_thanks",
"56": "pain_back_butt",
"57": "pain_breast",
"58": "pain_head",
"59": "pain_leg",
"60": "pain_lower_abdomen",
"61": "pain_muscle_joint",
"62": "pain_other",
"63": "pain_stomach",
"64": "periods",
"65": "pregnancy_black line",
"66": "pregnancy_clothing",
"67": "pregnancy_general",
"68": "pregnancy_weight_loss_gain",
"69": "pregnancy_working",
"70": "sleeping_position",
"71": "spitting",
"72": "stretchmarks",
"73": "survey_response",
"74": "swelling",
"75": "traditional_birth_attendant",
"76": "twins",
"77": "ultrasound",
"78": "urination_uti",
"79": "vaccination_immunization"
},
"initializer_range": 0.02,
"intermediate_size": 4096,
"label2id": {
"abortion": 0,
"ambulance": 1,
"anc_visit": 2,
"baby_acne_pimples": 3,
"baby_breathing": 4,
"baby_colic": 5,
"baby_constipation": 6,
"baby_cord": 7,
"baby_diarrhoea": 8,
"baby_fever_discharge": 9,
"baby_general": 10,
"baby_hiccups": 11,
"baby_jaundice": 12,
"baby_milestone_general": 13,
"baby_milestone_sitting": 14,
"baby_milestone_teething": 15,
"baby_poop": 16,
"baby_rash": 17,
"baby_sex_determination": 18,
"baby_spitting_up": 19,
"bleeding": 20,
"blood_group": 21,
"blood_pressure": 22,
"breastfeeding": 23,
"c_section": 24,
"constipation": 25,
"covid-19_vaccine": 26,
"danger_signs": 27,
"delivery_bag_items": 28,
"depression": 29,
"diarrhoea": 30,
"diet_nutrition": 31,
"discharge": 32,
"dizziness": 33,
"edd": 34,
"family_planning": 35,
"fatigue": 36,
"fetal_movement": 37,
"fever": 38,
"financial": 39,
"hb_level": 40,
"heart_rate_shortness_of_breath": 41,
"heartburn": 42,
"hiv": 43,
"ifas": 44,
"intercourse": 45,
"itching_body": 46,
"labour_signs": 47,
"linda_mama": 48,
"medication_general": 49,
"miscarriage_baby_loss": 50,
"mobility_travelling": 51,
"mosquito": 52,
"nausea_vomiting": 53,
"numbness": 54,
"ok_thanks": 55,
"pain_back_butt": 56,
"pain_breast": 57,
"pain_head": 58,
"pain_leg": 59,
"pain_lower_abdomen": 60,
"pain_muscle_joint": 61,
"pain_other": 62,
"pain_stomach": 63,
"periods": 64,
"pregnancy_black line": 65,
"pregnancy_clothing": 66,
"pregnancy_general": 67,
"pregnancy_weight_loss_gain": 68,
"pregnancy_working": 69,
"sleeping_position": 70,
"spitting": 71,
"stretchmarks": 72,
"survey_response": 73,
"swelling": 74,
"traditional_birth_attendant": 75,
"twins": 76,
"ultrasound": 77,
"urination_uti": 78,
"vaccination_immunization": 79
},
"layer_norm_eps": 1e-05,
"max_position_embeddings": 514,
"model_type": "xlm-roberta",
"num_attention_heads": 16,
"num_hidden_layers": 24,
"output_past": true,
"pad_token_id": 1,
"position_embedding_type": "absolute",
"problem_type": "single_label_classification",
"torch_dtype": "float32",
"transformers_version": "4.26.1",
"type_vocab_size": 1,
"use_cache": true,
"vocab_size": 250002
}