pii-small-en / config.json
smoh's picture
v1.4: epoch 3 best checkpoint (F1=0.889, 41 entity types, 241K training examples)
98ca2c3 verified
{
"architectures": [
"PiiNerModel"
],
"backbone": "microsoft/deberta-v3-xsmall",
"char_cnn_filters": [
50,
50,
50
],
"char_cnn_widths": [
3,
4,
5
],
"char_embed_dim": 50,
"char_vocab_size": 256,
"dropout": 0.1,
"id2label": {
"0": "O",
"1": "B-SSN",
"2": "I-SSN",
"3": "B-CREDIT_CARD",
"4": "I-CREDIT_CARD",
"5": "B-BANK_ACCOUNT",
"6": "I-BANK_ACCOUNT",
"7": "B-PASSPORT_NUMBER",
"8": "I-PASSPORT_NUMBER",
"9": "B-DRIVERS_LICENSE",
"10": "I-DRIVERS_LICENSE",
"11": "B-TAX_ID",
"12": "I-TAX_ID",
"13": "B-PERSON",
"14": "I-PERSON",
"15": "B-EMAIL",
"16": "I-EMAIL",
"17": "B-PHONE",
"18": "I-PHONE",
"19": "B-DATE_OF_BIRTH",
"20": "I-DATE_OF_BIRTH",
"21": "B-STREET_ADDRESS",
"22": "I-STREET_ADDRESS",
"23": "B-IP_ADDRESS",
"24": "I-IP_ADDRESS",
"25": "B-USERNAME",
"26": "I-USERNAME",
"27": "B-DATE",
"28": "I-DATE",
"29": "B-LOCATION",
"30": "I-LOCATION",
"31": "B-ORGANIZATION",
"32": "I-ORGANIZATION",
"33": "B-URL",
"34": "I-URL",
"35": "B-LICENSE_PLATE",
"36": "I-LICENSE_PLATE",
"37": "B-AGE",
"38": "I-AGE",
"39": "B-NATIONALITY",
"40": "I-NATIONALITY",
"41": "B-GENDER",
"42": "I-GENDER",
"43": "B-RELIGION",
"44": "I-RELIGION",
"45": "B-MARITAL_STATUS",
"46": "I-MARITAL_STATUS",
"47": "B-MEDICAL_RECORD",
"48": "I-MEDICAL_RECORD",
"49": "B-EMPLOYEE_ID",
"50": "I-EMPLOYEE_ID",
"51": "B-STUDENT_ID",
"52": "I-STUDENT_ID",
"53": "B-ACCOUNT_NUMBER",
"54": "I-ACCOUNT_NUMBER",
"55": "B-PIN",
"56": "I-PIN",
"57": "B-PASSWORD",
"58": "I-PASSWORD",
"59": "B-BIOMETRIC",
"60": "I-BIOMETRIC",
"61": "B-VEHICLE_ID",
"62": "I-VEHICLE_ID",
"63": "B-DEVICE_ID",
"64": "I-DEVICE_ID",
"65": "B-CRYPTO_WALLET",
"66": "I-CRYPTO_WALLET",
"67": "B-IBAN",
"68": "I-IBAN",
"69": "B-SWIFT_CODE",
"70": "I-SWIFT_CODE",
"71": "B-INSURANCE_NUMBER",
"72": "I-INSURANCE_NUMBER",
"73": "B-SALARY",
"74": "I-SALARY",
"75": "B-CRIMINAL_RECORD",
"76": "I-CRIMINAL_RECORD",
"77": "B-POLITICAL_AFFILIATION",
"78": "I-POLITICAL_AFFILIATION",
"79": "B-SEXUAL_ORIENTATION",
"80": "I-SEXUAL_ORIENTATION",
"81": "B-HEALTH_CONDITION",
"82": "I-HEALTH_CONDITION"
},
"label2id": {
"O": 0,
"B-SSN": 1,
"I-SSN": 2,
"B-CREDIT_CARD": 3,
"I-CREDIT_CARD": 4,
"B-BANK_ACCOUNT": 5,
"I-BANK_ACCOUNT": 6,
"B-PASSPORT_NUMBER": 7,
"I-PASSPORT_NUMBER": 8,
"B-DRIVERS_LICENSE": 9,
"I-DRIVERS_LICENSE": 10,
"B-TAX_ID": 11,
"I-TAX_ID": 12,
"B-PERSON": 13,
"I-PERSON": 14,
"B-EMAIL": 15,
"I-EMAIL": 16,
"B-PHONE": 17,
"I-PHONE": 18,
"B-DATE_OF_BIRTH": 19,
"I-DATE_OF_BIRTH": 20,
"B-STREET_ADDRESS": 21,
"I-STREET_ADDRESS": 22,
"B-IP_ADDRESS": 23,
"I-IP_ADDRESS": 24,
"B-USERNAME": 25,
"I-USERNAME": 26,
"B-DATE": 27,
"I-DATE": 28,
"B-LOCATION": 29,
"I-LOCATION": 30,
"B-ORGANIZATION": 31,
"I-ORGANIZATION": 32,
"B-URL": 33,
"I-URL": 34,
"B-LICENSE_PLATE": 35,
"I-LICENSE_PLATE": 36,
"B-AGE": 37,
"I-AGE": 38,
"B-NATIONALITY": 39,
"I-NATIONALITY": 40,
"B-GENDER": 41,
"I-GENDER": 42,
"B-RELIGION": 43,
"I-RELIGION": 44,
"B-MARITAL_STATUS": 45,
"I-MARITAL_STATUS": 46,
"B-MEDICAL_RECORD": 47,
"I-MEDICAL_RECORD": 48,
"B-EMPLOYEE_ID": 49,
"I-EMPLOYEE_ID": 50,
"B-STUDENT_ID": 51,
"I-STUDENT_ID": 52,
"B-ACCOUNT_NUMBER": 53,
"I-ACCOUNT_NUMBER": 54,
"B-PIN": 55,
"I-PIN": 56,
"B-PASSWORD": 57,
"I-PASSWORD": 58,
"B-BIOMETRIC": 59,
"I-BIOMETRIC": 60,
"B-VEHICLE_ID": 61,
"I-VEHICLE_ID": 62,
"B-DEVICE_ID": 63,
"I-DEVICE_ID": 64,
"B-CRYPTO_WALLET": 65,
"I-CRYPTO_WALLET": 66,
"B-IBAN": 67,
"I-IBAN": 68,
"B-SWIFT_CODE": 69,
"I-SWIFT_CODE": 70,
"B-INSURANCE_NUMBER": 71,
"I-INSURANCE_NUMBER": 72,
"B-SALARY": 73,
"I-SALARY": 74,
"B-CRIMINAL_RECORD": 75,
"I-CRIMINAL_RECORD": 76,
"B-POLITICAL_AFFILIATION": 77,
"I-POLITICAL_AFFILIATION": 78,
"B-SEXUAL_ORIENTATION": 79,
"I-SEXUAL_ORIENTATION": 80,
"B-HEALTH_CONDITION": 81,
"I-HEALTH_CONDITION": 82
},
"max_char_len": 20,
"model_type": "pii_ner",
"torch_dtype": "float32",
"transformers_version": "4.45.2",
"num_labels": 83
}