tcmtest / config.json
Cube's picture
Upload config.json
ab5ad7b
{
"_num_labels": 22,
"architectures": [
"BertForTokenClassification"
],
"model_type": "bert",
"label2id" : {
"PAD": "0",
"B-DRUG": "1",
"I-DRUG": "2",
"B-DRUG.INGREDIENT": "3",
"I-DRUG.INGREDIENT": "4",
"B-DRUG.GROUP": "5",
"I-DRUG.GROUP": "6",
"B-DRUG.TASTE": "7",
"I-DRUG.TASTE": "8",
"B-DRUG.EFFICACY": "9",
"I-DRUG.EFFICACY": "10",
"B-DISEASE": "11",
"I-DISEASE": "12",
"B-DISEASE.GROUP": "13",
"I-DISEASE.GROUP": "14",
"B-SYMPTOM": "15",
"I-SYMPTOM": "16",
"B-SYNDROME": "17",
"I-SYNDROME": "18",
"O": "19",
"[CLS]": "20",
"[SEP]": "21"},
"id2label" : {
"0": "PAD",
"1": "B-DRUG",
"2": "I-DRUG",
"3": "B-DRUG.INGREDIENT",
"4": "I-DRUG.INGREDIENT",
"5": "B-DRUG.GROUP",
"6": "I-DRUG.GROUP",
"7": "B-DRUG.TASTE",
"8": "I-DRUG.TASTE",
"9": "B-DRUG.EFFICACY",
"10": "I-DRUG.EFFICACY",
"11": "B-DISEASE",
"12": "I-DISEASE",
"13": "B-DISEASE.GROUP",
"14": "I-DISEASE.GROUP",
"15": "B-SYMPTOM",
"16": "I-SYMPTOM",
"17": "B-SYNDROME",
"18": "I-SYNDROME",
"19": "O",
"20": "[CLS]",
"21": "[SEP]"
},
"attention_probs_dropout_prob": 0.1,
"directionality": "bidi",
"hidden_act": "gelu",
"hidden_dropout_prob": 0.1,
"hidden_size": 768,
"initializer_range": 0.02,
"intermediate_size": 3072,
"max_position_embeddings": 512,
"num_attention_heads": 12,
"num_hidden_layers": 12,
"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": 21128
}