sblank's picture
Add labels to config.json
adbdd93
{
"architectures": [
"BertForSequenceClassification"
],
"attention_probs_dropout_prob": 0.1,
"directionality": "bidi",
"gradient_checkpointing": false,
"hidden_act": "gelu",
"hidden_dropout_prob": 0.1,
"hidden_size": 768,
"id2label": {
"0": "details.address",
"1": "tour.postcode.select",
"2": "tour.finish",
"3": "details.safeplace",
"4": "details.preferedNeighbour",
"5": "details.avoidNeighbour",
"6": "tour.job.collected",
"7": "no",
"8": "yes",
"9": "tour.start",
"10": "tour.details",
"11": "tour.job.signature",
"12": "tour.job.delivered",
"13": "select",
"14": "tour.job.safePlace",
"15": "safeplace",
"16": "navigate",
"17": "tour.job.carriedForward",
"18": "tour.job.failed",
"19": "help",
"20": "navigate.back",
"21": "undefined"
},
"initializer_range": 0.02,
"intermediate_size": 3072,
"label2id": {
"details.address": 0,
"tour.postcode.select": 1,
"tour.finish": 2,
"details.safeplace": 3,
"details.preferedNeighbour": 4,
"details.avoidNeighbour": 5,
"tour.job.collected": 6,
"no": 7,
"yes": 8,
"tour.start": 9,
"tour.details": 10,
"tour.job.signature": 11,
"tour.job.delivered": 12,
"select": 13,
"tour.job.safePlace": 14,
"safeplace": 15,
"navigate": 16,
"tour.job.carriedForward": 17,
"tour.job.failed": 18,
"help": 19,
"navigate.back": 20,
"undefined": 21
},
"layer_norm_eps": 1e-12,
"max_position_embeddings": 512,
"model_type": "bert",
"num_attention_heads": 12,
"num_hidden_layers": 12,
"pad_token_id": 0,
"pooler_fc_size": 768,
"pooler_num_attention_heads": 12,
"pooler_num_fc_layers": 3,
"pooler_size_per_head": 128,
"pooler_type": "first_token_transform",
"type_vocab_size": 2,
"vocab_size": 119547
}