MSLacerda's picture
MSLacerda/attribute_minig_mslacerda
2803ec7 verified
{
"_name_or_path": "distilbert/distilbert-base-uncased",
"activation": "gelu",
"architectures": [
"DistilBertForTokenClassification"
],
"attention_dropout": 0.1,
"dim": 768,
"dropout": 0.1,
"hidden_dim": 3072,
"id2label": {
"0": "coffee_brand",
"1": "coffee_item_form",
"2": "coffee_specialty",
"3": "coffee_flavor",
"4": "coffee_net_content",
"5": "coffee_pack_size",
"6": "coffee_roast_type",
"7": "coffee_caffeine_content",
"8": "coffee_machine_type",
"9": "coffee_container_type",
"10": "coffee_country",
"11": "shoes_brand",
"12": "shoes_gender",
"13": "shoes_model_name",
"14": "shoes_shoe_type",
"15": "shoes_color",
"16": "shoes_size",
"17": "shoes_material",
"18": "shoes_closure",
"19": "shoes_sport",
"20": "shoes_age",
"21": "shoes_waterproof",
"22": "toothbrush_brand",
"23": "toothbrush_age",
"24": "toothbrush_power_source",
"25": "toothbrush_cap",
"26": "toothbrush_color",
"27": "toothbrush_specialty",
"28": "toothbrush_operation_mode",
"29": "toothbrush_pack_size",
"30": "toothbrush_material",
"31": "toothbrush_firmness",
"32": "toothbrush_size",
"33": "toothbrush_animal",
"34": "conditioner_brand",
"35": "conditioner_item_form",
"36": "conditioner_gender",
"37": "conditioner_volume",
"38": "conditioner_specialty",
"39": "conditioner_pack_size",
"40": "conditioner_scent",
"41": "conditioner_benefit",
"42": "conditioner_hair_type",
"43": "conditioner_age",
"44": "vitamin_brand",
"45": "vitamin_supplement_type",
"46": "vitamin_net_content",
"47": "vitamin_item_form",
"48": "vitamin_health_benefit",
"49": "vitamin_pack_size",
"50": "vitamin_dosage",
"51": "vitamin_diet",
"52": "vitamin_flavor",
"53": "vitamin_gender",
"54": "vitamin_age",
"55": "vitamin_specialty",
"56": "vitamin_supply_size",
"57": "vitamin_administration_type",
"58": "vitamin_caffeine_content",
"59": "laundry_detergent_brand",
"60": "laundry_detergent_item_form",
"61": "laundry_detergent_net_content",
"62": "laundry_detergent_pack_size",
"63": "laundry_detergent_specialty",
"64": "laundry_detergent_scent",
"65": "laundry_detergent_load_size",
"66": "laundry_detergent_specific_uses",
"67": "laundry_detergent_color",
"68": "laundry_detergent_skin_type",
"69": "laundry_detergent_age",
"70": "tea_brand",
"71": "tea_tea_variety",
"72": "tea_flavor",
"73": "tea_pack_size",
"74": "tea_specialty",
"75": "tea_item_form",
"76": "tea_net_content",
"77": "tea_machine_type",
"78": "tea_country",
"79": "tea_caffeine_content",
"80": "tea_container_type",
"81": "sunscreen_brand",
"82": "sunscreen_item_form",
"83": "sunscreen_specialty",
"84": "sunscreen_protection_level",
"85": "sunscreen_net_content",
"86": "sunscreen_color",
"87": "sunscreen_age",
"88": "sunscreen_scent",
"89": "sunscreen_pack_size",
"90": "sunscreen_skin_type",
"91": "sunscreen_gender",
"92": "safety_mask_brand",
"93": "safety_mask_net_content",
"94": "safety_mask_color",
"95": "safety_mask_mask_type",
"96": "safety_mask_age_range",
"97": "safety_mask_reusability",
"98": "safety_mask_specialty",
"99": "safety_mask_material",
"100": "safety_mask_layer",
"101": "safety_mask_closure_type",
"102": "safety_mask_size",
"103": "safety_mask_package_type",
"104": "safety_mask_gender",
"105": "safety_mask_certified_grade",
"106": "safety_mask_country",
"107": "breakfast_cereal_brand",
"108": "breakfast_cereal_specialty",
"109": "breakfast_cereal_flavor",
"110": "breakfast_cereal_net_content",
"111": "breakfast_cereal_container",
"112": "breakfast_cereal_pack_size",
"113": "breakfast_cereal_sub_brand",
"114": "breakfast_cereal_cereal_type"
},
"initializer_range": 0.02,
"label2id": {
"breakfast_cereal_brand": 107,
"breakfast_cereal_cereal_type": 114,
"breakfast_cereal_container": 111,
"breakfast_cereal_flavor": 109,
"breakfast_cereal_net_content": 110,
"breakfast_cereal_pack_size": 112,
"breakfast_cereal_specialty": 108,
"breakfast_cereal_sub_brand": 113,
"coffee_brand": 0,
"coffee_caffeine_content": 7,
"coffee_container_type": 9,
"coffee_country": 10,
"coffee_flavor": 3,
"coffee_item_form": 1,
"coffee_machine_type": 8,
"coffee_net_content": 4,
"coffee_pack_size": 5,
"coffee_roast_type": 6,
"coffee_specialty": 2,
"conditioner_age": 43,
"conditioner_benefit": 41,
"conditioner_brand": 34,
"conditioner_gender": 36,
"conditioner_hair_type": 42,
"conditioner_item_form": 35,
"conditioner_pack_size": 39,
"conditioner_scent": 40,
"conditioner_specialty": 38,
"conditioner_volume": 37,
"laundry_detergent_age": 69,
"laundry_detergent_brand": 59,
"laundry_detergent_color": 67,
"laundry_detergent_item_form": 60,
"laundry_detergent_load_size": 65,
"laundry_detergent_net_content": 61,
"laundry_detergent_pack_size": 62,
"laundry_detergent_scent": 64,
"laundry_detergent_skin_type": 68,
"laundry_detergent_specialty": 63,
"laundry_detergent_specific_uses": 66,
"safety_mask_age_range": 96,
"safety_mask_brand": 92,
"safety_mask_certified_grade": 105,
"safety_mask_closure_type": 101,
"safety_mask_color": 94,
"safety_mask_country": 106,
"safety_mask_gender": 104,
"safety_mask_layer": 100,
"safety_mask_mask_type": 95,
"safety_mask_material": 99,
"safety_mask_net_content": 93,
"safety_mask_package_type": 103,
"safety_mask_reusability": 97,
"safety_mask_size": 102,
"safety_mask_specialty": 98,
"shoes_age": 20,
"shoes_brand": 11,
"shoes_closure": 18,
"shoes_color": 15,
"shoes_gender": 12,
"shoes_material": 17,
"shoes_model_name": 13,
"shoes_shoe_type": 14,
"shoes_size": 16,
"shoes_sport": 19,
"shoes_waterproof": 21,
"sunscreen_age": 87,
"sunscreen_brand": 81,
"sunscreen_color": 86,
"sunscreen_gender": 91,
"sunscreen_item_form": 82,
"sunscreen_net_content": 85,
"sunscreen_pack_size": 89,
"sunscreen_protection_level": 84,
"sunscreen_scent": 88,
"sunscreen_skin_type": 90,
"sunscreen_specialty": 83,
"tea_brand": 70,
"tea_caffeine_content": 79,
"tea_container_type": 80,
"tea_country": 78,
"tea_flavor": 72,
"tea_item_form": 75,
"tea_machine_type": 77,
"tea_net_content": 76,
"tea_pack_size": 73,
"tea_specialty": 74,
"tea_tea_variety": 71,
"toothbrush_age": 23,
"toothbrush_animal": 33,
"toothbrush_brand": 22,
"toothbrush_cap": 25,
"toothbrush_color": 26,
"toothbrush_firmness": 31,
"toothbrush_material": 30,
"toothbrush_operation_mode": 28,
"toothbrush_pack_size": 29,
"toothbrush_power_source": 24,
"toothbrush_size": 32,
"toothbrush_specialty": 27,
"vitamin_administration_type": 57,
"vitamin_age": 54,
"vitamin_brand": 44,
"vitamin_caffeine_content": 58,
"vitamin_diet": 51,
"vitamin_dosage": 50,
"vitamin_flavor": 52,
"vitamin_gender": 53,
"vitamin_health_benefit": 48,
"vitamin_item_form": 47,
"vitamin_net_content": 46,
"vitamin_pack_size": 49,
"vitamin_specialty": 55,
"vitamin_supplement_type": 45,
"vitamin_supply_size": 56
},
"max_position_embeddings": 512,
"model_type": "distilbert",
"n_heads": 12,
"n_layers": 6,
"pad_token_id": 0,
"qa_dropout": 0.1,
"seq_classif_dropout": 0.2,
"sinusoidal_pos_embds": false,
"tie_weights_": true,
"torch_dtype": "float32",
"transformers_version": "4.41.2",
"vocab_size": 30522
}