Text Classification
Transformers
ONNX
Safetensors
Spanish
English
bert
distilbert
multilingual
android
offline
Instructions to use Jesus2498/travel-intent-classifier with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use Jesus2498/travel-intent-classifier with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-classification", model="Jesus2498/travel-intent-classifier")# Load model directly from transformers import AutoModel model = AutoModel.from_pretrained("Jesus2498/travel-intent-classifier", device_map="auto") - Notebooks
- Google Colab
- Kaggle
Upload 5 files
#1
by Jesus2498 - opened
- metadata.json +356 -0
- training_args.bin +0 -0
metadata.json
ADDED
|
@@ -0,0 +1,356 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"label2id": {
|
| 3 |
+
"ATTRACTION": 0,
|
| 4 |
+
"BAR": 1,
|
| 5 |
+
"CAFE": 2,
|
| 6 |
+
"CINEMA": 3,
|
| 7 |
+
"FAST_FOOD": 4,
|
| 8 |
+
"FUEL": 5,
|
| 9 |
+
"GALLERY": 6,
|
| 10 |
+
"GARDEN": 7,
|
| 11 |
+
"HOSPITAL": 8,
|
| 12 |
+
"MALL": 9,
|
| 13 |
+
"MUSEUM": 10,
|
| 14 |
+
"NONE": 11,
|
| 15 |
+
"PARK": 12,
|
| 16 |
+
"PHARMACY": 13,
|
| 17 |
+
"PUB": 14,
|
| 18 |
+
"RESTAURANT": 15,
|
| 19 |
+
"SUPERMARKET": 16,
|
| 20 |
+
"THEATRE": 17,
|
| 21 |
+
"VIEWPOINT": 18
|
| 22 |
+
},
|
| 23 |
+
"id2label": {
|
| 24 |
+
"0": "ATTRACTION",
|
| 25 |
+
"1": "BAR",
|
| 26 |
+
"2": "CAFE",
|
| 27 |
+
"3": "CINEMA",
|
| 28 |
+
"4": "FAST_FOOD",
|
| 29 |
+
"5": "FUEL",
|
| 30 |
+
"6": "GALLERY",
|
| 31 |
+
"7": "GARDEN",
|
| 32 |
+
"8": "HOSPITAL",
|
| 33 |
+
"9": "MALL",
|
| 34 |
+
"10": "MUSEUM",
|
| 35 |
+
"11": "NONE",
|
| 36 |
+
"12": "PARK",
|
| 37 |
+
"13": "PHARMACY",
|
| 38 |
+
"14": "PUB",
|
| 39 |
+
"15": "RESTAURANT",
|
| 40 |
+
"16": "SUPERMARKET",
|
| 41 |
+
"17": "THEATRE",
|
| 42 |
+
"18": "VIEWPOINT"
|
| 43 |
+
},
|
| 44 |
+
"subcategory2id": {
|
| 45 |
+
"BURGER": 0,
|
| 46 |
+
"CHINESE": 1,
|
| 47 |
+
"GENERAL": 2,
|
| 48 |
+
"INDIAN": 3,
|
| 49 |
+
"ITALIAN": 4,
|
| 50 |
+
"JAPANESE": 5,
|
| 51 |
+
"KOREAN": 6,
|
| 52 |
+
"MEXICAN": 7,
|
| 53 |
+
"PIZZA": 8,
|
| 54 |
+
"SEAFOOD": 9,
|
| 55 |
+
"SPANISH": 10,
|
| 56 |
+
"STEAKHOUSE": 11,
|
| 57 |
+
"THAI": 12,
|
| 58 |
+
"VEGAN": 13,
|
| 59 |
+
"VEGETARIAN": 14
|
| 60 |
+
},
|
| 61 |
+
"id2subcategory": {
|
| 62 |
+
"0": "BURGER",
|
| 63 |
+
"1": "CHINESE",
|
| 64 |
+
"2": "GENERAL",
|
| 65 |
+
"3": "INDIAN",
|
| 66 |
+
"4": "ITALIAN",
|
| 67 |
+
"5": "JAPANESE",
|
| 68 |
+
"6": "KOREAN",
|
| 69 |
+
"7": "MEXICAN",
|
| 70 |
+
"8": "PIZZA",
|
| 71 |
+
"9": "SEAFOOD",
|
| 72 |
+
"10": "SPANISH",
|
| 73 |
+
"11": "STEAKHOUSE",
|
| 74 |
+
"12": "THAI",
|
| 75 |
+
"13": "VEGAN",
|
| 76 |
+
"14": "VEGETARIAN"
|
| 77 |
+
},
|
| 78 |
+
"restaurant_type2id": {
|
| 79 |
+
"AIR": 0,
|
| 80 |
+
"ALLERGY": 1,
|
| 81 |
+
"ANIMALS": 2,
|
| 82 |
+
"APPETIZERS": 3,
|
| 83 |
+
"ART": 4,
|
| 84 |
+
"BAKERY": 5,
|
| 85 |
+
"BALLET": 6,
|
| 86 |
+
"BBQ": 7,
|
| 87 |
+
"BEEF": 8,
|
| 88 |
+
"BEER": 9,
|
| 89 |
+
"BIRYANI": 10,
|
| 90 |
+
"BOTANICAL": 11,
|
| 91 |
+
"BOWLS": 12,
|
| 92 |
+
"BREAD": 13,
|
| 93 |
+
"BREAKFAST": 14,
|
| 94 |
+
"BURGERS": 15,
|
| 95 |
+
"BURRITOS": 16,
|
| 96 |
+
"CAR_WASH": 17,
|
| 97 |
+
"CHEESE": 18,
|
| 98 |
+
"CHICKEN": 19,
|
| 99 |
+
"CLASSIC": 20,
|
| 100 |
+
"COCKTAILS": 21,
|
| 101 |
+
"COFFEE": 22,
|
| 102 |
+
"COLD_FLU": 23,
|
| 103 |
+
"CRAFT_BEER": 24,
|
| 104 |
+
"CURRY": 25,
|
| 105 |
+
"DAIRY": 26,
|
| 106 |
+
"DESSERTS": 27,
|
| 107 |
+
"DIGESTIVE": 28,
|
| 108 |
+
"DIM_SUM": 29,
|
| 109 |
+
"DOGS": 30,
|
| 110 |
+
"DOUBLE": 31,
|
| 111 |
+
"DRINKS": 32,
|
| 112 |
+
"ELECTRONICS": 33,
|
| 113 |
+
"EMERGENCY": 34,
|
| 114 |
+
"ENCHILADAS": 35,
|
| 115 |
+
"ENTERTAINMENT": 36,
|
| 116 |
+
"EV_CHARGING": 37,
|
| 117 |
+
"EYE_CARE": 38,
|
| 118 |
+
"FAMILY": 39,
|
| 119 |
+
"FASHION": 40,
|
| 120 |
+
"FIRST_AID": 41,
|
| 121 |
+
"FISH": 42,
|
| 122 |
+
"FLOWERS": 43,
|
| 123 |
+
"FRIED_CHICKEN": 44,
|
| 124 |
+
"FROZEN": 45,
|
| 125 |
+
"FRUITS": 46,
|
| 126 |
+
"FUEL": 47,
|
| 127 |
+
"GENERAL": 48,
|
| 128 |
+
"GENERAL_MEDICINE": 49,
|
| 129 |
+
"GIN": 50,
|
| 130 |
+
"GOURMET": 51,
|
| 131 |
+
"GRILL": 52,
|
| 132 |
+
"HISTORIC": 53,
|
| 133 |
+
"HISTORY": 54,
|
| 134 |
+
"HOT_DOGS": 55,
|
| 135 |
+
"IMAX": 56,
|
| 136 |
+
"JAPANESE": 57,
|
| 137 |
+
"KEBAB": 58,
|
| 138 |
+
"KOREAN": 59,
|
| 139 |
+
"LIVE_SHOWS": 60,
|
| 140 |
+
"LOCAL_ART": 61,
|
| 141 |
+
"MATERNITY": 62,
|
| 142 |
+
"MEAT": 63,
|
| 143 |
+
"MEDICATION": 64,
|
| 144 |
+
"MEXICAN": 65,
|
| 145 |
+
"MODERN_ART": 66,
|
| 146 |
+
"MOVIES": 67,
|
| 147 |
+
"MUSICALS": 68,
|
| 148 |
+
"NATURAL_HISTORY": 69,
|
| 149 |
+
"NATURE": 70,
|
| 150 |
+
"NON_ALCOHOLIC": 71,
|
| 151 |
+
"NOODLES": 72,
|
| 152 |
+
"OPERA": 73,
|
| 153 |
+
"ORGANIC": 74,
|
| 154 |
+
"ORIGINAL_VERSION": 75,
|
| 155 |
+
"PAINKILLERS": 76,
|
| 156 |
+
"PANORAMIC": 77,
|
| 157 |
+
"PANTRY": 78,
|
| 158 |
+
"PASTA": 79,
|
| 159 |
+
"PASTRY": 80,
|
| 160 |
+
"PEDIATRICS": 81,
|
| 161 |
+
"PETS": 82,
|
| 162 |
+
"PHOTOGRAPHY": 83,
|
| 163 |
+
"PICNIC": 84,
|
| 164 |
+
"PIZZA": 85,
|
| 165 |
+
"PLAYGROUND": 86,
|
| 166 |
+
"PLAYS": 87,
|
| 167 |
+
"PORK": 88,
|
| 168 |
+
"POULTRY": 89,
|
| 169 |
+
"QUESADILLAS": 90,
|
| 170 |
+
"RAMEN": 91,
|
| 171 |
+
"RELAX": 92,
|
| 172 |
+
"RELIGIOUS": 93,
|
| 173 |
+
"RIBS": 94,
|
| 174 |
+
"RICE": 95,
|
| 175 |
+
"RISOTTO": 96,
|
| 176 |
+
"RUM": 97,
|
| 177 |
+
"SALADS": 98,
|
| 178 |
+
"SANDWICHES": 99,
|
| 179 |
+
"SCIENCE": 100,
|
| 180 |
+
"SEAFOOD": 101,
|
| 181 |
+
"SERVICES": 102,
|
| 182 |
+
"SEXUAL_HEALTH": 103,
|
| 183 |
+
"SHELLFISH": 104,
|
| 184 |
+
"SHOPPING": 105,
|
| 185 |
+
"SIDES": 106,
|
| 186 |
+
"SMASH": 107,
|
| 187 |
+
"SNACKS": 108,
|
| 188 |
+
"SOUPS": 109,
|
| 189 |
+
"SPECIAL": 110,
|
| 190 |
+
"SPORTS": 111,
|
| 191 |
+
"STEAKS": 112,
|
| 192 |
+
"STREET_FOOD": 113,
|
| 193 |
+
"SUNRISE_SUNSET": 114,
|
| 194 |
+
"SUN_PROTECTION": 115,
|
| 195 |
+
"SURGERY": 116,
|
| 196 |
+
"SUSHI": 117,
|
| 197 |
+
"TACOS": 118,
|
| 198 |
+
"TANDOORI": 119,
|
| 199 |
+
"TAPAS": 120,
|
| 200 |
+
"TEA": 121,
|
| 201 |
+
"TEMPURA": 122,
|
| 202 |
+
"TESTS": 123,
|
| 203 |
+
"THAI": 124,
|
| 204 |
+
"THREE_D": 125,
|
| 205 |
+
"TIKKA": 126,
|
| 206 |
+
"TOYS": 127,
|
| 207 |
+
"TRADITIONAL": 128,
|
| 208 |
+
"TRANSPORT": 129,
|
| 209 |
+
"TRAUMA": 130,
|
| 210 |
+
"URBAN": 131,
|
| 211 |
+
"VEGAN": 132,
|
| 212 |
+
"VEGETABLES": 133,
|
| 213 |
+
"VEGETARIAN": 134,
|
| 214 |
+
"WHISKY": 135,
|
| 215 |
+
"WINE": 136
|
| 216 |
+
},
|
| 217 |
+
"id2restaurant_type": {
|
| 218 |
+
"0": "AIR",
|
| 219 |
+
"1": "ALLERGY",
|
| 220 |
+
"2": "ANIMALS",
|
| 221 |
+
"3": "APPETIZERS",
|
| 222 |
+
"4": "ART",
|
| 223 |
+
"5": "BAKERY",
|
| 224 |
+
"6": "BALLET",
|
| 225 |
+
"7": "BBQ",
|
| 226 |
+
"8": "BEEF",
|
| 227 |
+
"9": "BEER",
|
| 228 |
+
"10": "BIRYANI",
|
| 229 |
+
"11": "BOTANICAL",
|
| 230 |
+
"12": "BOWLS",
|
| 231 |
+
"13": "BREAD",
|
| 232 |
+
"14": "BREAKFAST",
|
| 233 |
+
"15": "BURGERS",
|
| 234 |
+
"16": "BURRITOS",
|
| 235 |
+
"17": "CAR_WASH",
|
| 236 |
+
"18": "CHEESE",
|
| 237 |
+
"19": "CHICKEN",
|
| 238 |
+
"20": "CLASSIC",
|
| 239 |
+
"21": "COCKTAILS",
|
| 240 |
+
"22": "COFFEE",
|
| 241 |
+
"23": "COLD_FLU",
|
| 242 |
+
"24": "CRAFT_BEER",
|
| 243 |
+
"25": "CURRY",
|
| 244 |
+
"26": "DAIRY",
|
| 245 |
+
"27": "DESSERTS",
|
| 246 |
+
"28": "DIGESTIVE",
|
| 247 |
+
"29": "DIM_SUM",
|
| 248 |
+
"30": "DOGS",
|
| 249 |
+
"31": "DOUBLE",
|
| 250 |
+
"32": "DRINKS",
|
| 251 |
+
"33": "ELECTRONICS",
|
| 252 |
+
"34": "EMERGENCY",
|
| 253 |
+
"35": "ENCHILADAS",
|
| 254 |
+
"36": "ENTERTAINMENT",
|
| 255 |
+
"37": "EV_CHARGING",
|
| 256 |
+
"38": "EYE_CARE",
|
| 257 |
+
"39": "FAMILY",
|
| 258 |
+
"40": "FASHION",
|
| 259 |
+
"41": "FIRST_AID",
|
| 260 |
+
"42": "FISH",
|
| 261 |
+
"43": "FLOWERS",
|
| 262 |
+
"44": "FRIED_CHICKEN",
|
| 263 |
+
"45": "FROZEN",
|
| 264 |
+
"46": "FRUITS",
|
| 265 |
+
"47": "FUEL",
|
| 266 |
+
"48": "GENERAL",
|
| 267 |
+
"49": "GENERAL_MEDICINE",
|
| 268 |
+
"50": "GIN",
|
| 269 |
+
"51": "GOURMET",
|
| 270 |
+
"52": "GRILL",
|
| 271 |
+
"53": "HISTORIC",
|
| 272 |
+
"54": "HISTORY",
|
| 273 |
+
"55": "HOT_DOGS",
|
| 274 |
+
"56": "IMAX",
|
| 275 |
+
"57": "JAPANESE",
|
| 276 |
+
"58": "KEBAB",
|
| 277 |
+
"59": "KOREAN",
|
| 278 |
+
"60": "LIVE_SHOWS",
|
| 279 |
+
"61": "LOCAL_ART",
|
| 280 |
+
"62": "MATERNITY",
|
| 281 |
+
"63": "MEAT",
|
| 282 |
+
"64": "MEDICATION",
|
| 283 |
+
"65": "MEXICAN",
|
| 284 |
+
"66": "MODERN_ART",
|
| 285 |
+
"67": "MOVIES",
|
| 286 |
+
"68": "MUSICALS",
|
| 287 |
+
"69": "NATURAL_HISTORY",
|
| 288 |
+
"70": "NATURE",
|
| 289 |
+
"71": "NON_ALCOHOLIC",
|
| 290 |
+
"72": "NOODLES",
|
| 291 |
+
"73": "OPERA",
|
| 292 |
+
"74": "ORGANIC",
|
| 293 |
+
"75": "ORIGINAL_VERSION",
|
| 294 |
+
"76": "PAINKILLERS",
|
| 295 |
+
"77": "PANORAMIC",
|
| 296 |
+
"78": "PANTRY",
|
| 297 |
+
"79": "PASTA",
|
| 298 |
+
"80": "PASTRY",
|
| 299 |
+
"81": "PEDIATRICS",
|
| 300 |
+
"82": "PETS",
|
| 301 |
+
"83": "PHOTOGRAPHY",
|
| 302 |
+
"84": "PICNIC",
|
| 303 |
+
"85": "PIZZA",
|
| 304 |
+
"86": "PLAYGROUND",
|
| 305 |
+
"87": "PLAYS",
|
| 306 |
+
"88": "PORK",
|
| 307 |
+
"89": "POULTRY",
|
| 308 |
+
"90": "QUESADILLAS",
|
| 309 |
+
"91": "RAMEN",
|
| 310 |
+
"92": "RELAX",
|
| 311 |
+
"93": "RELIGIOUS",
|
| 312 |
+
"94": "RIBS",
|
| 313 |
+
"95": "RICE",
|
| 314 |
+
"96": "RISOTTO",
|
| 315 |
+
"97": "RUM",
|
| 316 |
+
"98": "SALADS",
|
| 317 |
+
"99": "SANDWICHES",
|
| 318 |
+
"100": "SCIENCE",
|
| 319 |
+
"101": "SEAFOOD",
|
| 320 |
+
"102": "SERVICES",
|
| 321 |
+
"103": "SEXUAL_HEALTH",
|
| 322 |
+
"104": "SHELLFISH",
|
| 323 |
+
"105": "SHOPPING",
|
| 324 |
+
"106": "SIDES",
|
| 325 |
+
"107": "SMASH",
|
| 326 |
+
"108": "SNACKS",
|
| 327 |
+
"109": "SOUPS",
|
| 328 |
+
"110": "SPECIAL",
|
| 329 |
+
"111": "SPORTS",
|
| 330 |
+
"112": "STEAKS",
|
| 331 |
+
"113": "STREET_FOOD",
|
| 332 |
+
"114": "SUNRISE_SUNSET",
|
| 333 |
+
"115": "SUN_PROTECTION",
|
| 334 |
+
"116": "SURGERY",
|
| 335 |
+
"117": "SUSHI",
|
| 336 |
+
"118": "TACOS",
|
| 337 |
+
"119": "TANDOORI",
|
| 338 |
+
"120": "TAPAS",
|
| 339 |
+
"121": "TEA",
|
| 340 |
+
"122": "TEMPURA",
|
| 341 |
+
"123": "TESTS",
|
| 342 |
+
"124": "THAI",
|
| 343 |
+
"125": "THREE_D",
|
| 344 |
+
"126": "TIKKA",
|
| 345 |
+
"127": "TOYS",
|
| 346 |
+
"128": "TRADITIONAL",
|
| 347 |
+
"129": "TRANSPORT",
|
| 348 |
+
"130": "TRAUMA",
|
| 349 |
+
"131": "URBAN",
|
| 350 |
+
"132": "VEGAN",
|
| 351 |
+
"133": "VEGETABLES",
|
| 352 |
+
"134": "VEGETARIAN",
|
| 353 |
+
"135": "WHISKY",
|
| 354 |
+
"136": "WINE"
|
| 355 |
+
}
|
| 356 |
+
}
|
training_args.bin
CHANGED
|
Binary files a/training_args.bin and b/training_args.bin differ
|
|
|