Train 10k + CRF for nemotron-pii-ready. F1: 0.9618

nemotron-pii-ready/TokenBased-BERT/config.json

@@ -0,0 +1,252 @@
+{
+  "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": "B-ACCOUNT_NUMBER",
+    "1": "B-AGE",
+    "2": "B-API_KEY",
+    "3": "B-BANK_ROUTING_NUMBER",
+    "4": "B-BIOMETRIC_IDENTIFIER",
+    "5": "B-BLOOD_TYPE",
+    "6": "B-CERTIFICATE_LICENSE_NUMBER",
+    "7": "B-CITY",
+    "8": "B-COMPANY_NAME",
+    "9": "B-COORDINATE",
+    "10": "B-COUNTRY",
+    "11": "B-COUNTY",
+    "12": "B-CREDIT_DEBIT_CARD",
+    "13": "B-CUSTOMER_ID",
+    "14": "B-CVV",
+    "15": "B-DATE",
+    "16": "B-DATE_OF_BIRTH",
+    "17": "B-DATE_TIME",
+    "18": "B-DEVICE_IDENTIFIER",
+    "19": "B-EDUCATION_LEVEL",
+    "20": "B-EMAIL",
+    "21": "B-EMPLOYEE_ID",
+    "22": "B-EMPLOYMENT_STATUS",
+    "23": "B-FAX_NUMBER",
+    "24": "B-FIRST_NAME",
+    "25": "B-GENDER",
+    "26": "B-HEALTH_PLAN_BENEFICIARY_NUMBER",
+    "27": "B-HTTP_COOKIE",
+    "28": "B-IPADDRESS",
+    "29": "B-LANGUAGE",
+    "30": "B-LAST_NAME",
+    "31": "B-LICENSE_PLATE",
+    "32": "B-MAC_ADDRESS",
+    "33": "B-MEDICAL_RECORD_NUMBER",
+    "34": "B-NATIONAL_ID",
+    "35": "B-OCCUPATION",
+    "36": "B-PASSWORD",
+    "37": "B-PHONE_NUMBER",
+    "38": "B-PIN",
+    "39": "B-POLITICAL_VIEW",
+    "40": "B-POSTCODE",
+    "41": "B-RACE_ETHNICITY",
+    "42": "B-RELIGIOUS_BELIEF",
+    "43": "B-SEXUALITY",
+    "44": "B-SSN/CCCD",
+    "45": "B-STATE",
+    "46": "B-STREET_ADDRESS",
+    "47": "B-SWIFT_BIC",
+    "48": "B-TAX_ID",
+    "49": "B-TIME",
+    "50": "B-UNIQUE_ID",
+    "51": "B-URL",
+    "52": "B-USER_NAME",
+    "53": "B-VEHICLE_IDENTIFIER",
+    "54": "I-ACCOUNT_NUMBER",
+    "55": "I-AGE",
+    "56": "I-API_KEY",
+    "57": "I-BANK_ROUTING_NUMBER",
+    "58": "I-BIOMETRIC_IDENTIFIER",
+    "59": "I-BLOOD_TYPE",
+    "60": "I-CERTIFICATE_LICENSE_NUMBER",
+    "61": "I-CITY",
+    "62": "I-COMPANY_NAME",
+    "63": "I-COORDINATE",
+    "64": "I-COUNTRY",
+    "65": "I-COUNTY",
+    "66": "I-CREDIT_DEBIT_CARD",
+    "67": "I-CUSTOMER_ID",
+    "68": "I-CVV",
+    "69": "I-DATE",
+    "70": "I-DATE_OF_BIRTH",
+    "71": "I-DATE_TIME",
+    "72": "I-DEVICE_IDENTIFIER",
+    "73": "I-EDUCATION_LEVEL",
+    "74": "I-EMAIL",
+    "75": "I-EMPLOYEE_ID",
+    "76": "I-EMPLOYMENT_STATUS",
+    "77": "I-FAX_NUMBER",
+    "78": "I-FIRST_NAME",
+    "79": "I-GENDER",
+    "80": "I-HEALTH_PLAN_BENEFICIARY_NUMBER",
+    "81": "I-HTTP_COOKIE",
+    "82": "I-IPADDRESS",
+    "83": "I-LANGUAGE",
+    "84": "I-LAST_NAME",
+    "85": "I-LICENSE_PLATE",
+    "86": "I-MAC_ADDRESS",
+    "87": "I-MEDICAL_RECORD_NUMBER",
+    "88": "I-NATIONAL_ID",
+    "89": "I-OCCUPATION",
+    "90": "I-PASSWORD",
+    "91": "I-PHONE_NUMBER",
+    "92": "I-PIN",
+    "93": "I-POLITICAL_VIEW",
+    "94": "I-POSTCODE",
+    "95": "I-RACE_ETHNICITY",
+    "96": "I-RELIGIOUS_BELIEF",
+    "97": "I-SEXUALITY",
+    "98": "I-SSN/CCCD",
+    "99": "I-STATE",
+    "100": "I-STREET_ADDRESS",
+    "101": "I-SWIFT_BIC",
+    "102": "I-TAX_ID",
+    "103": "I-TIME",
+    "104": "I-UNIQUE_ID",
+    "105": "I-URL",
+    "106": "I-USER_NAME",
+    "107": "I-VEHICLE_IDENTIFIER",
+    "108": "O"
+  },
+  "initializer_range": 0.02,
+  "intermediate_size": 3072,
+  "label2id": {
+    "B-ACCOUNT_NUMBER": 0,
+    "B-AGE": 1,
+    "B-API_KEY": 2,
+    "B-BANK_ROUTING_NUMBER": 3,
+    "B-BIOMETRIC_IDENTIFIER": 4,
+    "B-BLOOD_TYPE": 5,
+    "B-CERTIFICATE_LICENSE_NUMBER": 6,
+    "B-CITY": 7,
+    "B-COMPANY_NAME": 8,
+    "B-COORDINATE": 9,
+    "B-COUNTRY": 10,
+    "B-COUNTY": 11,
+    "B-CREDIT_DEBIT_CARD": 12,
+    "B-CUSTOMER_ID": 13,
+    "B-CVV": 14,
+    "B-DATE": 15,
+    "B-DATE_OF_BIRTH": 16,
+    "B-DATE_TIME": 17,
+    "B-DEVICE_IDENTIFIER": 18,
+    "B-EDUCATION_LEVEL": 19,
+    "B-EMAIL": 20,
+    "B-EMPLOYEE_ID": 21,
+    "B-EMPLOYMENT_STATUS": 22,
+    "B-FAX_NUMBER": 23,
+    "B-FIRST_NAME": 24,
+    "B-GENDER": 25,
+    "B-HEALTH_PLAN_BENEFICIARY_NUMBER": 26,
+    "B-HTTP_COOKIE": 27,
+    "B-IPADDRESS": 28,
+    "B-LANGUAGE": 29,
+    "B-LAST_NAME": 30,
+    "B-LICENSE_PLATE": 31,
+    "B-MAC_ADDRESS": 32,
+    "B-MEDICAL_RECORD_NUMBER": 33,
+    "B-NATIONAL_ID": 34,
+    "B-OCCUPATION": 35,
+    "B-PASSWORD": 36,
+    "B-PHONE_NUMBER": 37,
+    "B-PIN": 38,
+    "B-POLITICAL_VIEW": 39,
+    "B-POSTCODE": 40,
+    "B-RACE_ETHNICITY": 41,
+    "B-RELIGIOUS_BELIEF": 42,
+    "B-SEXUALITY": 43,
+    "B-SSN/CCCD": 44,
+    "B-STATE": 45,
+    "B-STREET_ADDRESS": 46,
+    "B-SWIFT_BIC": 47,
+    "B-TAX_ID": 48,
+    "B-TIME": 49,
+    "B-UNIQUE_ID": 50,
+    "B-URL": 51,
+    "B-USER_NAME": 52,
+    "B-VEHICLE_IDENTIFIER": 53,
+    "I-ACCOUNT_NUMBER": 54,
+    "I-AGE": 55,
+    "I-API_KEY": 56,
+    "I-BANK_ROUTING_NUMBER": 57,
+    "I-BIOMETRIC_IDENTIFIER": 58,
+    "I-BLOOD_TYPE": 59,
+    "I-CERTIFICATE_LICENSE_NUMBER": 60,
+    "I-CITY": 61,
+    "I-COMPANY_NAME": 62,
+    "I-COORDINATE": 63,
+    "I-COUNTRY": 64,
+    "I-COUNTY": 65,
+    "I-CREDIT_DEBIT_CARD": 66,
+    "I-CUSTOMER_ID": 67,
+    "I-CVV": 68,
+    "I-DATE": 69,
+    "I-DATE_OF_BIRTH": 70,
+    "I-DATE_TIME": 71,
+    "I-DEVICE_IDENTIFIER": 72,
+    "I-EDUCATION_LEVEL": 73,
+    "I-EMAIL": 74,
+    "I-EMPLOYEE_ID": 75,
+    "I-EMPLOYMENT_STATUS": 76,
+    "I-FAX_NUMBER": 77,
+    "I-FIRST_NAME": 78,
+    "I-GENDER": 79,
+    "I-HEALTH_PLAN_BENEFICIARY_NUMBER": 80,
+    "I-HTTP_COOKIE": 81,
+    "I-IPADDRESS": 82,
+    "I-LANGUAGE": 83,
+    "I-LAST_NAME": 84,
+    "I-LICENSE_PLATE": 85,
+    "I-MAC_ADDRESS": 86,
+    "I-MEDICAL_RECORD_NUMBER": 87,
+    "I-NATIONAL_ID": 88,
+    "I-OCCUPATION": 89,
+    "I-PASSWORD": 90,
+    "I-PHONE_NUMBER": 91,
+    "I-PIN": 92,
+    "I-POLITICAL_VIEW": 93,
+    "I-POSTCODE": 94,
+    "I-RACE_ETHNICITY": 95,
+    "I-RELIGIOUS_BELIEF": 96,
+    "I-SEXUALITY": 97,
+    "I-SSN/CCCD": 98,
+    "I-STATE": 99,
+    "I-STREET_ADDRESS": 100,
+    "I-SWIFT_BIC": 101,
+    "I-TAX_ID": 102,
+    "I-TIME": 103,
+    "I-UNIQUE_ID": 104,
+    "I-URL": 105,
+    "I-USER_NAME": 106,
+    "I-VEHICLE_IDENTIFIER": 107,
+    "O": 108
+  },
+  "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.56.0",
+  "type_vocab_size": 2,
+  "use_cache": false,
+  "vocab_size": 119547
+}

nemotron-pii-ready/TokenBased-BERT/model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:4649138c23283fe799aab9937ded575e6bb58db7787e7f039aaf963dd511e993
+oid sha256:6e3b1c7ee6eba638fe02ddb6377c45baa8b85edbb7c6817a5b9894ce0949fe00
 size 711821256

nemotron-pii-ready/TokenBased-BERT/training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:e8a60c4b5ef8d85a9f06594d38fa25eeb77e83dafcbfc5b502ca6b4157986510
+oid sha256:c42a27d9755be130f1e33b02c56d544b73c29c68df62f18464429769b372eab0
 size 5777