korean_disease_ner / config.json
keonju's picture
Training in progress, epoch 1
00c75cd
raw
history blame
1.81 kB
{
"_name_or_path": "klue/bert-base",
"architectures": [
"BertForTokenClassification"
],
"attention_probs_dropout_prob": 0.1,
"classifier_dropout": null,
"hidden_act": "gelu",
"hidden_dropout_prob": 0.1,
"hidden_size": 768,
"id2label": {
"0": "O",
"1": "PS_NAME",
"2": "FD_MEDICINE",
"3": "TR_MEDICINE",
"4": "AF_TRANSPORT",
"5": "OGG_MEDICINE",
"6": "LCP_COUNTRY",
"7": "TMM_DISEASE",
"8": "TM_CELL_TISSUE_ORGAN",
"9": "TMM_DRUG",
"10": "CV_RELATION",
"11": "CV_FOOD",
"12": "CV_DRINK",
"13": "CV_CLOTHING",
"14": "DT_DAY",
"15": "DT_WEEK",
"16": "DT_MONTH",
"17": "DT_YEAR",
"18": "DT_SEASON",
"19": "TI_DURATION",
"20": "TI_HOUR",
"21": "TI_MINUTE",
"22": "TI_SECOND",
"23": "QT_AGE",
"24": "QT_MAN_COUNT",
"25": "QT_PHONE"
},
"initializer_range": 0.02,
"intermediate_size": 3072,
"label2id": {
"AF_TRANSPORT": 4,
"CV_CLOTHING": 13,
"CV_DRINK": 12,
"CV_FOOD": 11,
"CV_RELATION": 10,
"DT_DAY": 14,
"DT_MONTH": 16,
"DT_SEASON": 18,
"DT_WEEK": 15,
"DT_YEAR": 17,
"FD_MEDICINE": 2,
"LCP_COUNTRY": 6,
"O": 0,
"OGG_MEDICINE": 5,
"PS_NAME": 1,
"QT_AGE": 23,
"QT_MAN_COUNT": 24,
"QT_PHONE": 25,
"TI_DURATION": 19,
"TI_HOUR": 20,
"TI_MINUTE": 21,
"TI_SECOND": 22,
"TMM_DISEASE": 7,
"TMM_DRUG": 9,
"TM_CELL_TISSUE_ORGAN": 8,
"TR_MEDICINE": 3
},
"layer_norm_eps": 1e-12,
"max_position_embeddings": 512,
"model_type": "bert",
"num_attention_heads": 12,
"num_hidden_layers": 12,
"pad_token_id": 0,
"position_embedding_type": "absolute",
"torch_dtype": "float32",
"transformers_version": "4.26.0",
"type_vocab_size": 2,
"use_cache": true,
"vocab_size": 32000
}