multilingual_DialogPII_NER / crf_config.json
deryaerman's picture
Add FLERT model files and update README with FLERT inference code
0e344f5
Raw
History Blame Contribute Delete
2.36 kB
{
"base_model_name": "jhu-clsp/mmBERT-base",
"num_labels": 48,
"id2label": {
"0": "O",
"1": "B-CITY",
"2": "B-CODE",
"3": "B-CODE_PHONE",
"4": "B-CODE_URL",
"5": "B-DATETIME",
"6": "B-DATETIME_AGE",
"7": "B-DISEASE",
"8": "B-LOC_CITY",
"9": "B-LOC_COUNTRY",
"10": "B-LOC_DISTRICT",
"11": "B-LOC_HOUSENUMBER",
"12": "B-LOC_NAME",
"13": "B-LOC_OTHER",
"14": "B-LOC_STREET",
"15": "B-LOC_ZIP",
"16": "B-MISC",
"17": "B-NAME",
"18": "B-ORG",
"19": "B-PERSON",
"20": "B-PERSON_EMAIL",
"21": "B-PERSON_SOCIAL_RELATION",
"22": "B-PRODUCT",
"23": "B-PROFESSION",
"24": "B-QUANTITY",
"25": "I-CITY",
"26": "I-CODE",
"27": "I-CODE_PHONE",
"28": "I-DATETIME",
"29": "I-DATETIME_AGE",
"30": "I-DISEASE",
"31": "I-LOC_CITY",
"32": "I-LOC_COUNTRY",
"33": "I-LOC_DISTRICT",
"34": "I-LOC_HOUSENUMBER",
"35": "I-LOC_NAME",
"36": "I-LOC_OTHER",
"37": "I-LOC_STREET",
"38": "I-LOC_ZIP",
"39": "I-MISC",
"40": "I-NAME",
"41": "I-ORG",
"42": "I-PERSON",
"43": "I-PERSON_EMAIL",
"44": "I-PERSON_SOCIAL_RELATION",
"45": "I-PRODUCT",
"46": "I-PROFESSION",
"47": "I-QUANTITY"
},
"label2id": {
"O": 0,
"B-CITY": 1,
"B-CODE": 2,
"B-CODE_PHONE": 3,
"B-CODE_URL": 4,
"B-DATETIME": 5,
"B-DATETIME_AGE": 6,
"B-DISEASE": 7,
"B-LOC_CITY": 8,
"B-LOC_COUNTRY": 9,
"B-LOC_DISTRICT": 10,
"B-LOC_HOUSENUMBER": 11,
"B-LOC_NAME": 12,
"B-LOC_OTHER": 13,
"B-LOC_STREET": 14,
"B-LOC_ZIP": 15,
"B-MISC": 16,
"B-NAME": 17,
"B-ORG": 18,
"B-PERSON": 19,
"B-PERSON_EMAIL": 20,
"B-PERSON_SOCIAL_RELATION": 21,
"B-PRODUCT": 22,
"B-PROFESSION": 23,
"B-QUANTITY": 24,
"I-CITY": 25,
"I-CODE": 26,
"I-CODE_PHONE": 27,
"I-DATETIME": 28,
"I-DATETIME_AGE": 29,
"I-DISEASE": 30,
"I-LOC_CITY": 31,
"I-LOC_COUNTRY": 32,
"I-LOC_DISTRICT": 33,
"I-LOC_HOUSENUMBER": 34,
"I-LOC_NAME": 35,
"I-LOC_OTHER": 36,
"I-LOC_STREET": 37,
"I-LOC_ZIP": 38,
"I-MISC": 39,
"I-NAME": 40,
"I-ORG": 41,
"I-PERSON": 42,
"I-PERSON_EMAIL": 43,
"I-PERSON_SOCIAL_RELATION": 44,
"I-PRODUCT": 45,
"I-PROFESSION": 46,
"I-QUANTITY": 47
},
"use_crf": true
}