File size: 3,967 Bytes
9ad9505 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 |
{
"_name_or_path": "./pt_save_pretrained_ner",
"activation": "gelu",
"architectures": [
"DistilBertForTokenClassification"
],
"attention_dropout": 0.1,
"dim": 768,
"dropout": 0.1,
"hidden_dim": 3072,
"id2label": {
"0": "B-location_name",
"1": "I-artist",
"2": "B-state",
"3": "B-object_location_type",
"4": "B-artist",
"5": "I-served_dish",
"6": "I-album",
"7": "B-track",
"8": "I-playlist_owner",
"9": "I-restaurant_type",
"10": "B-spatial_relation",
"11": "I-spatial_relation",
"12": "B-object_select",
"13": "B-condition_temperature",
"14": "B-geographic_poi",
"15": "I-facility",
"16": "B-playlist",
"17": "B-facility",
"18": "I-object_name",
"19": "I-geographic_poi",
"20": "B-movie_type",
"21": "B-restaurant_name",
"22": "I-music_item",
"23": "I-poi",
"24": "I-cuisine",
"25": "I-movie_name",
"26": "I-sort",
"27": "B-cuisine",
"28": "I-country",
"29": "B-service",
"30": "I-restaurant_name",
"31": "B-album",
"32": "B-entity_name",
"33": "I-service",
"34": "I-movie_type",
"35": "B-served_dish",
"36": "B-best_rating",
"37": "B-rating_value",
"38": "I-object_part_of_series_type",
"39": "I-genre",
"40": "I-object_location_type",
"41": "B-restaurant_type",
"42": "B-object_part_of_series_type",
"43": "O",
"44": "I-city",
"45": "B-music_item",
"46": "I-party_size_description",
"47": "B-city",
"48": "B-playlist_owner",
"49": "B-country",
"50": "B-condition_description",
"51": "B-year",
"52": "I-object_type",
"53": "B-sort",
"54": "B-timeRange",
"55": "B-current_location",
"56": "I-state",
"57": "I-object_select",
"58": "B-object_name",
"59": "I-track",
"60": "B-rating_unit",
"61": "I-location_name",
"62": "B-party_size_number",
"63": "I-current_location",
"64": "B-poi",
"65": "B-movie_name",
"66": "I-entity_name",
"67": "B-object_type",
"68": "B-party_size_description",
"69": "B-genre",
"70": "I-timeRange",
"71": "I-playlist"
},
"initializer_range": 0.02,
"label2id": {
"LABEL_0": 0,
"LABEL_1": 1,
"LABEL_10": 10,
"LABEL_11": 11,
"LABEL_12": 12,
"LABEL_13": 13,
"LABEL_14": 14,
"LABEL_15": 15,
"LABEL_16": 16,
"LABEL_17": 17,
"LABEL_18": 18,
"LABEL_19": 19,
"LABEL_2": 2,
"LABEL_20": 20,
"LABEL_21": 21,
"LABEL_22": 22,
"LABEL_23": 23,
"LABEL_24": 24,
"LABEL_25": 25,
"LABEL_26": 26,
"LABEL_27": 27,
"LABEL_28": 28,
"LABEL_29": 29,
"LABEL_3": 3,
"LABEL_30": 30,
"LABEL_31": 31,
"LABEL_32": 32,
"LABEL_33": 33,
"LABEL_34": 34,
"LABEL_35": 35,
"LABEL_36": 36,
"LABEL_37": 37,
"LABEL_38": 38,
"LABEL_39": 39,
"LABEL_4": 4,
"LABEL_40": 40,
"LABEL_41": 41,
"LABEL_42": 42,
"LABEL_43": 43,
"LABEL_44": 44,
"LABEL_45": 45,
"LABEL_46": 46,
"LABEL_47": 47,
"LABEL_48": 48,
"LABEL_49": 49,
"LABEL_5": 5,
"LABEL_50": 50,
"LABEL_51": 51,
"LABEL_52": 52,
"LABEL_53": 53,
"LABEL_54": 54,
"LABEL_55": 55,
"LABEL_56": 56,
"LABEL_57": 57,
"LABEL_58": 58,
"LABEL_59": 59,
"LABEL_6": 6,
"LABEL_60": 60,
"LABEL_61": 61,
"LABEL_62": 62,
"LABEL_63": 63,
"LABEL_64": 64,
"LABEL_65": 65,
"LABEL_66": 66,
"LABEL_67": 67,
"LABEL_68": 68,
"LABEL_69": 69,
"LABEL_7": 7,
"LABEL_70": 70,
"LABEL_71": 71,
"LABEL_8": 8,
"LABEL_9": 9
},
"max_position_embeddings": 512,
"model_type": "distilbert",
"n_heads": 12,
"n_layers": 6,
"output_past": true,
"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.24.0",
"vocab_size": 28996
}
|