pii-model-indic-v2 / config.json
apxrv's picture
Upload folder using huggingface_hub
9b8aa2b verified
{
"architectures": [
"AlbertForTokenClassification"
],
"attention_probs_dropout_prob": 0,
"bos_token_id": 2,
"classifier_dropout_prob": 0.1,
"down_scale_factor": 1,
"dtype": "float32",
"embedding_size": 128,
"eos_token_id": 3,
"gap_size": 0,
"hidden_act": "gelu",
"hidden_dropout_prob": 0,
"hidden_size": 768,
"id2label": {
"0": "O",
"1": "B-ACCOUNTNUMBER",
"2": "I-ACCOUNTNUMBER",
"3": "B-FIRSTNAME",
"4": "I-FIRSTNAME",
"5": "B-ACCOUNTNAME",
"6": "I-ACCOUNTNAME",
"7": "B-PHONENUMBER",
"8": "I-PHONENUMBER",
"9": "B-CREDITCARDCVV",
"10": "I-CREDITCARDCVV",
"11": "B-CREDITCARDISSUER",
"12": "I-CREDITCARDISSUER",
"13": "B-PREFIX",
"14": "I-PREFIX",
"15": "B-LASTNAME",
"16": "I-LASTNAME",
"17": "B-AMOUNT",
"18": "I-AMOUNT",
"19": "B-DATE",
"20": "I-DATE",
"21": "B-DOB",
"22": "I-DOB",
"23": "B-COMPANYNAME",
"24": "I-COMPANYNAME",
"25": "B-BUILDINGNUMBER",
"26": "I-BUILDINGNUMBER",
"27": "B-STREET",
"28": "I-STREET",
"29": "B-SECONDARYADDRESS",
"30": "I-SECONDARYADDRESS",
"31": "B-STATE",
"32": "I-STATE",
"33": "B-EMAIL",
"34": "I-EMAIL",
"35": "B-CITY",
"36": "I-CITY",
"37": "B-CREDITCARDNUMBER",
"38": "I-CREDITCARDNUMBER",
"39": "B-SSN",
"40": "I-SSN",
"41": "B-URL",
"42": "I-URL",
"43": "B-USERNAME",
"44": "I-USERNAME",
"45": "B-PASSWORD",
"46": "I-PASSWORD",
"47": "B-COUNTY",
"48": "I-COUNTY",
"49": "B-PIN",
"50": "I-PIN",
"51": "B-MIDDLENAME",
"52": "I-MIDDLENAME",
"53": "B-IBAN",
"54": "I-IBAN",
"55": "B-GENDER",
"56": "I-GENDER",
"57": "B-AGE",
"58": "I-AGE",
"59": "B-ZIPCODE",
"60": "I-ZIPCODE",
"61": "B-SEX",
"62": "I-SEX"
},
"initializer_range": 0.02,
"inner_group_num": 1,
"intermediate_size": 3072,
"label2id": {
"B-ACCOUNTNAME": 5,
"B-ACCOUNTNUMBER": 1,
"B-AGE": 57,
"B-AMOUNT": 17,
"B-BUILDINGNUMBER": 25,
"B-CITY": 35,
"B-COMPANYNAME": 23,
"B-COUNTY": 47,
"B-CREDITCARDCVV": 9,
"B-CREDITCARDISSUER": 11,
"B-CREDITCARDNUMBER": 37,
"B-DATE": 19,
"B-DOB": 21,
"B-EMAIL": 33,
"B-FIRSTNAME": 3,
"B-GENDER": 55,
"B-IBAN": 53,
"B-LASTNAME": 15,
"B-MIDDLENAME": 51,
"B-PASSWORD": 45,
"B-PHONENUMBER": 7,
"B-PIN": 49,
"B-PREFIX": 13,
"B-SECONDARYADDRESS": 29,
"B-SEX": 61,
"B-SSN": 39,
"B-STATE": 31,
"B-STREET": 27,
"B-URL": 41,
"B-USERNAME": 43,
"B-ZIPCODE": 59,
"I-ACCOUNTNAME": 6,
"I-ACCOUNTNUMBER": 2,
"I-AGE": 58,
"I-AMOUNT": 18,
"I-BUILDINGNUMBER": 26,
"I-CITY": 36,
"I-COMPANYNAME": 24,
"I-COUNTY": 48,
"I-CREDITCARDCVV": 10,
"I-CREDITCARDISSUER": 12,
"I-CREDITCARDNUMBER": 38,
"I-DATE": 20,
"I-DOB": 22,
"I-EMAIL": 34,
"I-FIRSTNAME": 4,
"I-GENDER": 56,
"I-IBAN": 54,
"I-LASTNAME": 16,
"I-MIDDLENAME": 52,
"I-PASSWORD": 46,
"I-PHONENUMBER": 8,
"I-PIN": 50,
"I-PREFIX": 14,
"I-SECONDARYADDRESS": 30,
"I-SEX": 62,
"I-SSN": 40,
"I-STATE": 32,
"I-STREET": 28,
"I-URL": 42,
"I-USERNAME": 44,
"I-ZIPCODE": 60,
"O": 0
},
"layer_norm_eps": 1e-12,
"max_position_embeddings": 512,
"model_type": "albert",
"net_structure_type": 0,
"num_attention_heads": 12,
"num_hidden_groups": 1,
"num_hidden_layers": 12,
"num_memory_blocks": 0,
"pad_token_id": 0,
"position_embedding_type": "absolute",
"tie_word_embeddings": true,
"transformers_version": "5.3.0",
"type_vocab_size": 2,
"use_cache": false,
"vocab_size": 200000
}