roberta-pii-ner-en / config.json
rm0013's picture
Model files
17e302d verified
{
"add_cross_attention": false,
"architectures": [
"RobertaForTokenClassification"
],
"attention_probs_dropout_prob": 0.1,
"bos_token_id": 0,
"classifier_dropout": null,
"dtype": "float32",
"eos_token_id": 2,
"hidden_act": "gelu",
"hidden_dropout_prob": 0.1,
"hidden_size": 768,
"id2label": {
"0": "O",
"1": "B-ACCOUNTNAME",
"2": "B-ADDRESS",
"3": "B-AGE",
"4": "B-AMOUNT",
"5": "B-API_KEY",
"6": "B-BANK_ACCOUNT",
"7": "B-BANK_ROUTING",
"8": "B-BIC",
"9": "B-BITCOINADDRESS",
"10": "B-COMPANYNAME",
"11": "B-CREDITCARDISSUER",
"12": "B-CREDIT_CARD",
"13": "B-CREDIT_CARD_CVV",
"14": "B-CREDIT_CARD_EXPIRY",
"15": "B-CURRENCY",
"16": "B-CURRENCYCODE",
"17": "B-CURRENCYNAME",
"18": "B-CURRENCYSYMBOL",
"19": "B-DATE",
"20": "B-DATE_OF_BIRTH",
"21": "B-DEVICE_ID",
"22": "B-DRIVER_LICENSE",
"23": "B-EMAIL",
"24": "B-ETHEREUMADDRESS",
"25": "B-EYECOLOR",
"26": "B-GENDER",
"27": "B-GPS_COORDINATES",
"28": "B-HEIGHT",
"29": "B-IP_ADDRESS",
"30": "B-JOBAREA",
"31": "B-JOBTITLE",
"32": "B-JOBTYPE",
"33": "B-LITECOINADDRESS",
"34": "B-MASKEDNUMBER",
"35": "B-NEARBYGPSCOORDINATE",
"36": "B-ORDINALDIRECTION",
"37": "B-ORGANIZATION",
"38": "B-PASSPORT_NUMBER",
"39": "B-PASSWORD",
"40": "B-PERSON_NAME",
"41": "B-PHONEIMEI",
"42": "B-PHONE_NUMBER",
"43": "B-PIN",
"44": "B-SECONDARYADDRESS",
"45": "B-SEX",
"46": "B-SSN",
"47": "B-TIME",
"48": "B-URL",
"49": "B-USERAGENT",
"50": "B-USERNAME",
"51": "B-VEHICLEVIN",
"52": "B-VEHICLEVRM",
"53": "B-VEHICLE_ID",
"54": "I-ACCOUNTNAME",
"55": "I-ADDRESS",
"56": "I-AGE",
"57": "I-AMOUNT",
"58": "I-COMPANYNAME",
"59": "I-CURRENCY",
"60": "I-CURRENCYNAME",
"61": "I-DATE",
"62": "I-DATE_OF_BIRTH",
"63": "I-EYECOLOR",
"64": "I-GENDER",
"65": "I-GPS_COORDINATES",
"66": "I-HEIGHT",
"67": "I-JOBTITLE",
"68": "I-ORGANIZATION",
"69": "I-PERSON_NAME",
"70": "I-PHONE_NUMBER",
"71": "I-SECONDARYADDRESS",
"72": "I-SSN",
"73": "I-TIME",
"74": "I-USERAGENT"
},
"initializer_range": 0.02,
"intermediate_size": 3072,
"is_decoder": false,
"label2id": {
"B-ACCOUNTNAME": 1,
"B-ADDRESS": 2,
"B-AGE": 3,
"B-AMOUNT": 4,
"B-API_KEY": 5,
"B-BANK_ACCOUNT": 6,
"B-BANK_ROUTING": 7,
"B-BIC": 8,
"B-BITCOINADDRESS": 9,
"B-COMPANYNAME": 10,
"B-CREDITCARDISSUER": 11,
"B-CREDIT_CARD": 12,
"B-CREDIT_CARD_CVV": 13,
"B-CREDIT_CARD_EXPIRY": 14,
"B-CURRENCY": 15,
"B-CURRENCYCODE": 16,
"B-CURRENCYNAME": 17,
"B-CURRENCYSYMBOL": 18,
"B-DATE": 19,
"B-DATE_OF_BIRTH": 20,
"B-DEVICE_ID": 21,
"B-DRIVER_LICENSE": 22,
"B-EMAIL": 23,
"B-ETHEREUMADDRESS": 24,
"B-EYECOLOR": 25,
"B-GENDER": 26,
"B-GPS_COORDINATES": 27,
"B-HEIGHT": 28,
"B-IP_ADDRESS": 29,
"B-JOBAREA": 30,
"B-JOBTITLE": 31,
"B-JOBTYPE": 32,
"B-LITECOINADDRESS": 33,
"B-MASKEDNUMBER": 34,
"B-NEARBYGPSCOORDINATE": 35,
"B-ORDINALDIRECTION": 36,
"B-ORGANIZATION": 37,
"B-PASSPORT_NUMBER": 38,
"B-PASSWORD": 39,
"B-PERSON_NAME": 40,
"B-PHONEIMEI": 41,
"B-PHONE_NUMBER": 42,
"B-PIN": 43,
"B-SECONDARYADDRESS": 44,
"B-SEX": 45,
"B-SSN": 46,
"B-TIME": 47,
"B-URL": 48,
"B-USERAGENT": 49,
"B-USERNAME": 50,
"B-VEHICLEVIN": 51,
"B-VEHICLEVRM": 52,
"B-VEHICLE_ID": 53,
"I-ACCOUNTNAME": 54,
"I-ADDRESS": 55,
"I-AGE": 56,
"I-AMOUNT": 57,
"I-COMPANYNAME": 58,
"I-CURRENCY": 59,
"I-CURRENCYNAME": 60,
"I-DATE": 61,
"I-DATE_OF_BIRTH": 62,
"I-EYECOLOR": 63,
"I-GENDER": 64,
"I-GPS_COORDINATES": 65,
"I-HEIGHT": 66,
"I-JOBTITLE": 67,
"I-ORGANIZATION": 68,
"I-PERSON_NAME": 69,
"I-PHONE_NUMBER": 70,
"I-SECONDARYADDRESS": 71,
"I-SSN": 72,
"I-TIME": 73,
"I-USERAGENT": 74,
"O": 0
},
"layer_norm_eps": 1e-05,
"max_position_embeddings": 514,
"model_type": "roberta",
"num_attention_heads": 12,
"num_hidden_layers": 12,
"pad_token_id": 1,
"tie_word_embeddings": true,
"transformers_version": "5.0.0",
"type_vocab_size": 1,
"use_cache": false,
"vocab_size": 50265
}