finetuned_model / config.json
Iris314's picture
Training in progress, epoch 1
9a0f004 verified
{
"activation": "gelu",
"architectures": [
"DistilBertForSequenceClassification"
],
"attention_dropout": 0.1,
"dim": 768,
"dropout": 0.1,
"dtype": "float32",
"hidden_dim": 3072,
"id2label": {
"0": "Italian-American",
"1": "Middle Eastern",
"2": "Korean",
"3": "Asian",
"4": "Bakery",
"5": "Hungarian",
"6": "Vegan",
"7": "BBQ",
"8": "Asian Fusion",
"9": "Japanese",
"10": "Italian",
"11": "Caribbean",
"12": "Coffee",
"13": "Belgian",
"14": "American",
"15": "Thai",
"16": "Mexican",
"17": "Irish",
"18": "Chinese",
"19": "Seafood",
"20": "Argentinian",
"21": "Dessert",
"22": "German",
"23": "Pizza",
"24": "Steakhouse",
"25": "Uzbek",
"26": "Peruvian",
"27": "Food Hall",
"28": "Deli",
"29": "Caf\u00e9",
"30": "Cocktail Bar",
"31": "French"
},
"initializer_range": 0.02,
"label2id": {
"American": 14,
"Argentinian": 20,
"Asian": 3,
"Asian Fusion": 8,
"BBQ": 7,
"Bakery": 4,
"Belgian": 13,
"Caf\u00e9": 29,
"Caribbean": 11,
"Chinese": 18,
"Cocktail Bar": 30,
"Coffee": 12,
"Deli": 28,
"Dessert": 21,
"Food Hall": 27,
"French": 31,
"German": 22,
"Hungarian": 5,
"Irish": 17,
"Italian": 10,
"Italian-American": 0,
"Japanese": 9,
"Korean": 2,
"Mexican": 16,
"Middle Eastern": 1,
"Peruvian": 26,
"Pizza": 23,
"Seafood": 19,
"Steakhouse": 24,
"Thai": 15,
"Uzbek": 25,
"Vegan": 6
},
"max_position_embeddings": 512,
"model_type": "distilbert",
"n_heads": 12,
"n_layers": 6,
"pad_token_id": 0,
"problem_type": "single_label_classification",
"qa_dropout": 0.1,
"seq_classif_dropout": 0.2,
"sinusoidal_pos_embds": false,
"tie_weights_": true,
"transformers_version": "4.56.1",
"vocab_size": 30522
}