Upload folder using huggingface_hub

nemotron/TokenBased-BERT/classification_report.json

@@ -0,0 +1,338 @@
+{
+  "ACCOUNT_NUMBER": {
+    "precision": 0.9657,
+    "recall": 0.9602,
+    "f1-score": 0.963,
+    "support": 176
+  },
+  "AGE": {
+    "precision": 0.9479,
+    "recall": 0.9681,
+    "f1-score": 0.9579,
+    "support": 94
+  },
+  "API_KEY": {
+    "precision": 1.0,
+    "recall": 0.9643,
+    "f1-score": 0.9818,
+    "support": 28
+  },
+  "BANK_ROUTING_NUMBER": {
+    "precision": 0.98,
+    "recall": 0.9703,
+    "f1-score": 0.9751,
+    "support": 101
+  },
+  "BIOMETRIC_IDENTIFIER": {
+    "precision": 0.9905,
+    "recall": 1.0,
+    "f1-score": 0.9952,
+    "support": 104
+  },
+  "BLOOD_TYPE": {
+    "precision": 1.0,
+    "recall": 1.0,
+    "f1-score": 1.0,
+    "support": 53
+  },
+  "CERTIFICATE_LICENSE_NUMBER": {
+    "precision": 1.0,
+    "recall": 0.9722,
+    "f1-score": 0.9859,
+    "support": 72
+  },
+  "CITY": {
+    "precision": 0.974,
+    "recall": 0.9677,
+    "f1-score": 0.9709,
+    "support": 155
+  },
+  "COMPANY_NAME": {
+    "precision": 0.9709,
+    "recall": 0.9471,
+    "f1-score": 0.9589,
+    "support": 529
+  },
+  "COORDINATE": {
+    "precision": 0.9787,
+    "recall": 1.0,
+    "f1-score": 0.9892,
+    "support": 92
+  },
+  "COUNTRY": {
+    "precision": 0.9854,
+    "recall": 0.9902,
+    "f1-score": 0.9878,
+    "support": 204
+  },
+  "COUNTY": {
+    "precision": 1.0,
+    "recall": 0.9461,
+    "f1-score": 0.9723,
+    "support": 167
+  },
+  "CREDIT_DEBIT_CARD": {
+    "precision": 1.0,
+    "recall": 0.9708,
+    "f1-score": 0.9852,
+    "support": 137
+  },
+  "CUSTOMER_ID": {
+    "precision": 0.9884,
+    "recall": 0.9942,
+    "f1-score": 0.9913,
+    "support": 171
+  },
+  "CVV": {
+    "precision": 1.0,
+    "recall": 0.9455,
+    "f1-score": 0.972,
+    "support": 55
+  },
+  "DATE": {
+    "precision": 0.969,
+    "recall": 0.9173,
+    "f1-score": 0.9425,
+    "support": 750
+  },
+  "DATE_OF_BIRTH": {
+    "precision": 0.9766,
+    "recall": 1.0,
+    "f1-score": 0.9881,
+    "support": 125
+  },
+  "DATE_TIME": {
+    "precision": 0.931,
+    "recall": 0.8526,
+    "f1-score": 0.8901,
+    "support": 95
+  },
+  "DEVICE_IDENTIFIER": {
+    "precision": 1.0,
+    "recall": 1.0,
+    "f1-score": 1.0,
+    "support": 18
+  },
+  "EDUCATION_LEVEL": {
+    "precision": 0.9649,
+    "recall": 0.9016,
+    "f1-score": 0.9322,
+    "support": 122
+  },
+  "EMAIL": {
+    "precision": 0.9974,
+    "recall": 0.9796,
+    "f1-score": 0.9884,
+    "support": 392
+  },
+  "EMPLOYEE_ID": {
+    "precision": 0.974,
+    "recall": 0.9375,
+    "f1-score": 0.9554,
+    "support": 80
+  },
+  "EMPLOYMENT_STATUS": {
+    "precision": 0.9278,
+    "recall": 0.9278,
+    "f1-score": 0.9278,
+    "support": 97
+  },
+  "FAX_NUMBER": {
+    "precision": 0.8971,
+    "recall": 0.9385,
+    "f1-score": 0.9173,
+    "support": 65
+  },
+  "FIRST_NAME": {
+    "precision": 0.9964,
+    "recall": 0.9808,
+    "f1-score": 0.9885,
+    "support": 835
+  },
+  "GENDER": {
+    "precision": 0.9208,
+    "recall": 0.9789,
+    "f1-score": 0.949,
+    "support": 95
+  },
+  "HEALTH_PLAN_BENEFICIARY_NUMBER": {
+    "precision": 1.0,
+    "recall": 0.9495,
+    "f1-score": 0.9741,
+    "support": 99
+  },
+  "HTTP_COOKIE": {
+    "precision": 0.9825,
+    "recall": 0.9492,
+    "f1-score": 0.9655,
+    "support": 59
+  },
+  "IPADDRESS": {
+    "precision": 1.0,
+    "recall": 0.975,
+    "f1-score": 0.9873,
+    "support": 80
+  },
+  "LANGUAGE": {
+    "precision": 0.9524,
+    "recall": 0.9677,
+    "f1-score": 0.96,
+    "support": 62
+  },
+  "LAST_NAME": {
+    "precision": 0.9874,
+    "recall": 0.9945,
+    "f1-score": 0.9909,
+    "support": 550
+  },
+  "LICENSE_PLATE": {
+    "precision": 0.9804,
+    "recall": 1.0,
+    "f1-score": 0.9901,
+    "support": 50
+  },
+  "MAC_ADDRESS": {
+    "precision": 1.0,
+    "recall": 0.9583,
+    "f1-score": 0.9787,
+    "support": 48
+  },
+  "MEDICAL_RECORD_NUMBER": {
+    "precision": 0.9829,
+    "recall": 1.0,
+    "f1-score": 0.9914,
+    "support": 115
+  },
+  "OCCUPATION": {
+    "precision": 0.9042,
+    "recall": 0.5975,
+    "f1-score": 0.7195,
+    "support": 395
+  },
+  "PASSWORD": {
+    "precision": 0.9844,
+    "recall": 0.9403,
+    "f1-score": 0.9618,
+    "support": 67
+  },
+  "PHONE_NUMBER": {
+    "precision": 0.9757,
+    "recall": 0.9663,
+    "f1-score": 0.971,
+    "support": 208
+  },
+  "PIN": {
+    "precision": 0.931,
+    "recall": 0.8438,
+    "f1-score": 0.8852,
+    "support": 64
+  },
+  "POLITICAL_VIEW": {
+    "precision": 0.8696,
+    "recall": 0.9524,
+    "f1-score": 0.9091,
+    "support": 63
+  },
+  "POSTCODE": {
+    "precision": 0.9516,
+    "recall": 0.9833,
+    "f1-score": 0.9672,
+    "support": 60
+  },
+  "RACE_ETHNICITY": {
+    "precision": 1.0,
+    "recall": 0.9906,
+    "f1-score": 0.9953,
+    "support": 106
+  },
+  "RELIGIOUS_BELIEF": {
+    "precision": 0.88,
+    "recall": 0.8,
+    "f1-score": 0.8381,
+    "support": 55
+  },
+  "SEXUALITY": {
+    "precision": 0.8684,
+    "recall": 1.0,
+    "f1-score": 0.9296,
+    "support": 33
+  },
+  "SSN": {
+    "precision": 0.9892,
+    "recall": 1.0,
+    "f1-score": 0.9946,
+    "support": 92
+  },
+  "STATE": {
+    "precision": 0.9683,
+    "recall": 0.9531,
+    "f1-score": 0.9606,
+    "support": 192
+  },
+  "STREET_ADDRESS": {
+    "precision": 0.9882,
+    "recall": 0.9941,
+    "f1-score": 0.9912,
+    "support": 169
+  },
+  "SWIFT_BIC": {
+    "precision": 0.9574,
+    "recall": 1.0,
+    "f1-score": 0.9783,
+    "support": 45
+  },
+  "TAX_ID": {
+    "precision": 0.9091,
+    "recall": 0.9091,
+    "f1-score": 0.9091,
+    "support": 11
+  },
+  "TIME": {
+    "precision": 0.9492,
+    "recall": 0.7249,
+    "f1-score": 0.822,
+    "support": 309
+  },
+  "UNIQUE_ID": {
+    "precision": 0.8333,
+    "recall": 0.9091,
+    "f1-score": 0.8696,
+    "support": 11
+  },
+  "URL": {
+    "precision": 1.0,
+    "recall": 0.9736,
+    "f1-score": 0.9866,
+    "support": 379
+  },
+  "USER_NAME": {
+    "precision": 0.9699,
+    "recall": 0.9348,
+    "f1-score": 0.952,
+    "support": 138
+  },
+  "VEHICLE_IDENTIFIER": {
+    "precision": 1.0,
+    "recall": 1.0,
+    "f1-score": 1.0,
+    "support": 51
+  },
+  "OVERALL_MICRO": {
+    "precision": 0.9744,
+    "recall": 0.9367,
+    "f1-score": 0.9552,
+    "support": 8323
+  },
+  "OVERALL_MACRO": {
+    "precision": 0.9651,
+    "recall": 0.9487,
+    "f1-score": 0.9556,
+    "support": 8323
+  },
+  "OVERALL_WEIGHTED": {
+    "precision": 0.9733,
+    "recall": 0.9367,
+    "f1-score": 0.9527,
+    "support": 8323
+  }
+}

nemotron/TokenBased-BERT/config.json

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

nemotron/TokenBased-BERT/model.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a0d6ea66ae71453dbb57f2c3f9cacb33a8cfeabd101effec81be6d8b65229a74
+size 435925196

nemotron/TokenBased-BERT/tokenizer_config.json

@@ -0,0 +1,16 @@
+{
+  "add_prefix_space": true,
+  "backend": "tokenizers",
+  "cls_token": "[CLS]",
+  "do_lower_case": true,
+  "is_local": false,
+  "local_files_only": false,
+  "mask_token": "[MASK]",
+  "model_max_length": 512,
+  "pad_token": "[PAD]",
+  "sep_token": "[SEP]",
+  "strip_accents": null,
+  "tokenize_chinese_chars": true,
+  "tokenizer_class": "BertTokenizer",
+  "unk_token": "[UNK]"
+}

nemotron/TokenBased-BERT/training_args.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:0177829520d218929a88f6460404a19799f7d93ba89980a581d9fc457b65241e
+size 5265