pii-ner-nemotron / config.json
scanpatch's picture
Upload folder using huggingface_hub
99a8c78 verified
{
"architectures": [
"XLMRobertaForTokenClassification"
],
"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": 1024,
"id2label": {
"0": "O",
"1": "B-address",
"2": "I-address",
"3": "B-address_apartment",
"4": "I-address_apartment",
"5": "B-address_building",
"6": "I-address_building",
"7": "B-address_city",
"8": "I-address_city",
"9": "B-address_country",
"10": "I-address_country",
"11": "B-address_district",
"12": "I-address_district",
"13": "B-address_geolocation",
"14": "I-address_geolocation",
"15": "B-address_house",
"16": "I-address_house",
"17": "B-address_postal_code",
"18": "I-address_postal_code",
"19": "B-address_region",
"20": "I-address_region",
"21": "B-address_street",
"22": "I-address_street",
"23": "B-date",
"24": "I-date",
"25": "B-document_number",
"26": "I-document_number",
"27": "B-email",
"28": "I-email",
"29": "B-first_name",
"30": "I-first_name",
"31": "B-ip",
"32": "I-ip",
"33": "B-last_name",
"34": "I-last_name",
"35": "B-middle_name",
"36": "I-middle_name",
"37": "B-military_individual_number",
"38": "I-military_individual_number",
"39": "B-mobile_phone",
"40": "I-mobile_phone",
"41": "B-name",
"42": "I-name",
"43": "B-name_initials",
"44": "I-name_initials",
"45": "B-nickname",
"46": "I-nickname",
"47": "B-organization",
"48": "I-organization",
"49": "B-snils",
"50": "I-snils",
"51": "B-tin",
"52": "I-tin",
"53": "B-vehicle_number",
"54": "I-vehicle_number"
},
"initializer_range": 0.02,
"intermediate_size": 4096,
"label2id": {
"B-address": 1,
"B-address_apartment": 3,
"B-address_building": 5,
"B-address_city": 7,
"B-address_country": 9,
"B-address_district": 11,
"B-address_geolocation": 13,
"B-address_house": 15,
"B-address_postal_code": 17,
"B-address_region": 19,
"B-address_street": 21,
"B-date": 23,
"B-document_number": 25,
"B-email": 27,
"B-first_name": 29,
"B-ip": 31,
"B-last_name": 33,
"B-middle_name": 35,
"B-military_individual_number": 37,
"B-mobile_phone": 39,
"B-name": 41,
"B-name_initials": 43,
"B-nickname": 45,
"B-organization": 47,
"B-snils": 49,
"B-tin": 51,
"B-vehicle_number": 53,
"I-address": 2,
"I-address_apartment": 4,
"I-address_building": 6,
"I-address_city": 8,
"I-address_country": 10,
"I-address_district": 12,
"I-address_geolocation": 14,
"I-address_house": 16,
"I-address_postal_code": 18,
"I-address_region": 20,
"I-address_street": 22,
"I-date": 24,
"I-document_number": 26,
"I-email": 28,
"I-first_name": 30,
"I-ip": 32,
"I-last_name": 34,
"I-middle_name": 36,
"I-military_individual_number": 38,
"I-mobile_phone": 40,
"I-name": 42,
"I-name_initials": 44,
"I-nickname": 46,
"I-organization": 48,
"I-snils": 50,
"I-tin": 52,
"I-vehicle_number": 54,
"O": 0
},
"layer_norm_eps": 1e-05,
"max_position_embeddings": 514,
"model_type": "xlm-roberta",
"num_attention_heads": 16,
"num_hidden_layers": 24,
"output_past": true,
"pad_token_id": 1,
"position_embedding_type": "absolute",
"transformers_version": "4.57.3",
"type_vocab_size": 1,
"use_cache": true,
"vocab_size": 250002
}