pii-ner / config.json
bldeaw's picture
Upload folder using huggingface_hub
6f7ca21 verified
Raw
History Blame Contribute Delete
3.23 kB
{
"architectures": [
"BertForTokenClassification"
],
"attention_probs_dropout_prob": 0.1,
"classifier_dropout": null,
"directionality": "bidi",
"dtype": "float32",
"hidden_act": "gelu",
"hidden_dropout_prob": 0.1,
"hidden_size": 768,
"id2label": {
"0": "O",
"1": "B-FULL_NAME",
"2": "I-FULL_NAME",
"3": "B-DATE_OF_BIRTH",
"4": "I-DATE_OF_BIRTH",
"5": "B-AGE",
"6": "I-AGE",
"7": "B-GENDER",
"8": "I-GENDER",
"9": "B-NATIONALITY",
"10": "I-NATIONALITY",
"11": "B-ETHNICITY",
"12": "I-ETHNICITY",
"13": "B-RELIGION",
"14": "I-RELIGION",
"15": "B-MARITAL_STATUS",
"16": "I-MARITAL_STATUS",
"17": "B-PHONE",
"18": "I-PHONE",
"19": "B-HOME_PHONE",
"20": "I-HOME_PHONE",
"21": "B-EMAIL",
"22": "I-EMAIL",
"23": "B-ADDRESS",
"24": "I-ADDRESS",
"25": "B-NATIONAL_ID",
"26": "I-NATIONAL_ID",
"27": "B-PASSPORT_NUMBER",
"28": "I-PASSPORT_NUMBER",
"29": "B-DRIVER_LICENSE",
"30": "I-DRIVER_LICENSE",
"31": "B-SOCIAL_SECURITY_NUMBER",
"32": "I-SOCIAL_SECURITY_NUMBER",
"33": "B-TAX_ID",
"34": "I-TAX_ID",
"35": "B-BANK_ACCOUNT",
"36": "I-BANK_ACCOUNT",
"37": "B-CREDIT_CARD",
"38": "I-CREDIT_CARD",
"39": "B-INCOME",
"40": "I-INCOME",
"41": "B-OCCUPATION",
"42": "I-OCCUPATION",
"43": "B-WORKPLACE",
"44": "I-WORKPLACE",
"45": "B-JOB_TITLE",
"46": "I-JOB_TITLE",
"47": "B-EDUCATION_LEVEL",
"48": "I-EDUCATION_LEVEL"
},
"initializer_range": 0.02,
"intermediate_size": 3072,
"label2id": {
"B-ADDRESS": 23,
"B-AGE": 5,
"B-BANK_ACCOUNT": 35,
"B-CREDIT_CARD": 37,
"B-DATE_OF_BIRTH": 3,
"B-DRIVER_LICENSE": 29,
"B-EDUCATION_LEVEL": 47,
"B-EMAIL": 21,
"B-ETHNICITY": 11,
"B-FULL_NAME": 1,
"B-GENDER": 7,
"B-HOME_PHONE": 19,
"B-INCOME": 39,
"B-JOB_TITLE": 45,
"B-MARITAL_STATUS": 15,
"B-NATIONALITY": 9,
"B-NATIONAL_ID": 25,
"B-OCCUPATION": 41,
"B-PASSPORT_NUMBER": 27,
"B-PHONE": 17,
"B-RELIGION": 13,
"B-SOCIAL_SECURITY_NUMBER": 31,
"B-TAX_ID": 33,
"B-WORKPLACE": 43,
"I-ADDRESS": 24,
"I-AGE": 6,
"I-BANK_ACCOUNT": 36,
"I-CREDIT_CARD": 38,
"I-DATE_OF_BIRTH": 4,
"I-DRIVER_LICENSE": 30,
"I-EDUCATION_LEVEL": 48,
"I-EMAIL": 22,
"I-ETHNICITY": 12,
"I-FULL_NAME": 2,
"I-GENDER": 8,
"I-HOME_PHONE": 20,
"I-INCOME": 40,
"I-JOB_TITLE": 46,
"I-MARITAL_STATUS": 16,
"I-NATIONALITY": 10,
"I-NATIONAL_ID": 26,
"I-OCCUPATION": 42,
"I-PASSPORT_NUMBER": 28,
"I-PHONE": 18,
"I-RELIGION": 14,
"I-SOCIAL_SECURITY_NUMBER": 32,
"I-TAX_ID": 34,
"I-WORKPLACE": 44,
"O": 0
},
"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",
"position_embedding_type": "absolute",
"transformers_version": "4.57.1",
"type_vocab_size": 2,
"use_cache": true,
"vocab_size": 119547
}