|
{ |
|
"_name_or_path": "AutoTrain", |
|
"_num_labels": 66, |
|
"architectures": [ |
|
"BertForSequenceClassification" |
|
], |
|
"attention_probs_dropout_prob": 0.1, |
|
"classifier_dropout": null, |
|
"gradient_checkpointing": false, |
|
"hidden_act": "gelu", |
|
"hidden_dropout_prob": 0.1, |
|
"hidden_size": 768, |
|
"id2label": { |
|
"0": "Afghan Restaurant", |
|
"1": "American Restaurant", |
|
"2": "Arepa Restaurant", |
|
"3": "Armenian Restaurant", |
|
"4": "Asian Restaurant", |
|
"5": "BBQ Joint", |
|
"6": "Bagel Shop", |
|
"7": "Bakery", |
|
"8": "Breakfast Spot", |
|
"9": "Buffet", |
|
"10": "Burger Joint", |
|
"11": "Caf\u00e9", |
|
"12": "Chaat Place", |
|
"13": "Chinese Restaurant", |
|
"14": "Coffee Shop", |
|
"15": "Comfort Food Restaurant", |
|
"16": "Creperie", |
|
"17": "Dessert Shop", |
|
"18": "Dim Sum Restaurant", |
|
"19": "Diner", |
|
"20": "Donut Shop", |
|
"21": "Egyptian Restaurant", |
|
"22": "English Restaurant", |
|
"23": "Falafel Restaurant", |
|
"24": "Fast Food Restaurant", |
|
"25": "Fish & Chips Shop", |
|
"26": "Food Stand", |
|
"27": "Food Truck", |
|
"28": "French Restaurant", |
|
"29": "Fried Chicken Joint", |
|
"30": "Greek Restaurant", |
|
"31": "Halal Restaurant", |
|
"32": "Health Food Store", |
|
"33": "Hot Dog Joint", |
|
"34": "Indian Restaurant", |
|
"35": "Indonesian Restaurant", |
|
"36": "Iraqi Restaurant", |
|
"37": "Italian Restaurant", |
|
"38": "Japanese Restaurant", |
|
"39": "Juice Bar", |
|
"40": "Kebab Restaurant", |
|
"41": "Lebanese Restaurant", |
|
"42": "Lounge", |
|
"43": "Mediterranean Restaurant", |
|
"44": "Mexican Restaurant", |
|
"45": "Middle Eastern Restaurant", |
|
"46": "Moroccan Restaurant", |
|
"47": "New American Restaurant", |
|
"48": "Noodle House", |
|
"49": "Pakistani Restaurant", |
|
"50": "Persian Restaurant", |
|
"51": "Pizza Place", |
|
"52": "Poke Place", |
|
"53": "Restaurant", |
|
"54": "Salad Place", |
|
"55": "Sandwich Place", |
|
"56": "Seafood Restaurant", |
|
"57": "Shawarma Place", |
|
"58": "Snack Place", |
|
"59": "Steakhouse", |
|
"60": "Sushi Restaurant", |
|
"61": "Taco Place", |
|
"62": "Tea Room", |
|
"63": "Thai Restaurant", |
|
"64": "Turkish Restaurant", |
|
"65": "Wings Joint" |
|
}, |
|
"initializer_range": 0.02, |
|
"intermediate_size": 3072, |
|
"label2id": { |
|
"Afghan Restaurant": 0, |
|
"American Restaurant": 1, |
|
"Arepa Restaurant": 2, |
|
"Armenian Restaurant": 3, |
|
"Asian Restaurant": 4, |
|
"BBQ Joint": 5, |
|
"Bagel Shop": 6, |
|
"Bakery": 7, |
|
"Breakfast Spot": 8, |
|
"Buffet": 9, |
|
"Burger Joint": 10, |
|
"Caf\u00e9": 11, |
|
"Chaat Place": 12, |
|
"Chinese Restaurant": 13, |
|
"Coffee Shop": 14, |
|
"Comfort Food Restaurant": 15, |
|
"Creperie": 16, |
|
"Dessert Shop": 17, |
|
"Dim Sum Restaurant": 18, |
|
"Diner": 19, |
|
"Donut Shop": 20, |
|
"Egyptian Restaurant": 21, |
|
"English Restaurant": 22, |
|
"Falafel Restaurant": 23, |
|
"Fast Food Restaurant": 24, |
|
"Fish & Chips Shop": 25, |
|
"Food Stand": 26, |
|
"Food Truck": 27, |
|
"French Restaurant": 28, |
|
"Fried Chicken Joint": 29, |
|
"Greek Restaurant": 30, |
|
"Halal Restaurant": 31, |
|
"Health Food Store": 32, |
|
"Hot Dog Joint": 33, |
|
"Indian Restaurant": 34, |
|
"Indonesian Restaurant": 35, |
|
"Iraqi Restaurant": 36, |
|
"Italian Restaurant": 37, |
|
"Japanese Restaurant": 38, |
|
"Juice Bar": 39, |
|
"Kebab Restaurant": 40, |
|
"Lebanese Restaurant": 41, |
|
"Lounge": 42, |
|
"Mediterranean Restaurant": 43, |
|
"Mexican Restaurant": 44, |
|
"Middle Eastern Restaurant": 45, |
|
"Moroccan Restaurant": 46, |
|
"New American Restaurant": 47, |
|
"Noodle House": 48, |
|
"Pakistani Restaurant": 49, |
|
"Persian Restaurant": 50, |
|
"Pizza Place": 51, |
|
"Poke Place": 52, |
|
"Restaurant": 53, |
|
"Salad Place": 54, |
|
"Sandwich Place": 55, |
|
"Seafood Restaurant": 56, |
|
"Shawarma Place": 57, |
|
"Snack Place": 58, |
|
"Steakhouse": 59, |
|
"Sushi Restaurant": 60, |
|
"Taco Place": 61, |
|
"Tea Room": 62, |
|
"Thai Restaurant": 63, |
|
"Turkish Restaurant": 64, |
|
"Wings Joint": 65 |
|
}, |
|
"layer_norm_eps": 1e-12, |
|
"max_length": 128, |
|
"max_position_embeddings": 512, |
|
"model_type": "bert", |
|
"num_attention_heads": 12, |
|
"num_hidden_layers": 12, |
|
"output_past": true, |
|
"pad_token_id": 0, |
|
"padding": "max_length", |
|
"position_embedding_type": "absolute", |
|
"problem_type": "single_label_classification", |
|
"torch_dtype": "float32", |
|
"transformers_version": "4.20.0", |
|
"type_vocab_size": 2, |
|
"use_cache": true, |
|
"vocab_size": 32000 |
|
} |
|
|