Train 10k + CRF for gretel-pii-ready. F1: 0.9576

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

@@ -0,0 +1,200 @@
+{
+  "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-ADDRESS",
+    "2": "B-API_KEY",
+    "3": "B-BANK_ROUTING_NUMBER",
+    "4": "B-BIOMETRIC_IDENTIFIER",
+    "5": "B-CERTIFICATE_LICENSE_NUMBER",
+    "6": "B-CITY",
+    "7": "B-COMPANY_NAME",
+    "8": "B-COORDINATE",
+    "9": "B-COUNTRY",
+    "10": "B-CREDIT_CARD_NUMBER",
+    "11": "B-CUSTOMER_ID",
+    "12": "B-CVV",
+    "13": "B-DATE",
+    "14": "B-DATE_OF_BIRTH",
+    "15": "B-DATE_TIME",
+    "16": "B-DEVICE_IDENTIFIER",
+    "17": "B-EMAIL",
+    "18": "B-EMPLOYEE_ID",
+    "19": "B-FIRST_NAME",
+    "20": "B-HEALTH_PLAN_BENEFICIARY_NUMBER",
+    "21": "B-IPADDRESS",
+    "22": "B-LAST_NAME",
+    "23": "B-LICENSE_PLATE",
+    "24": "B-MEDICAL_RECORD_NUMBER",
+    "25": "B-NAME",
+    "26": "B-NATIONAL_ID",
+    "27": "B-PASSWORD",
+    "28": "B-PHONE_NUMBER",
+    "29": "B-PIN",
+    "30": "B-POSTCODE",
+    "31": "B-SSN/CCCD",
+    "32": "B-STATE",
+    "33": "B-STREET_ADDRESS",
+    "34": "B-SWIFT_BIC",
+    "35": "B-TAX_ID",
+    "36": "B-TIME",
+    "37": "B-UNIQUE_IDENTIFIER",
+    "38": "B-URL",
+    "39": "B-USER_NAME",
+    "40": "B-VEHICLE_IDENTIFIER",
+    "41": "I-ACCOUNT_NUMBER",
+    "42": "I-ADDRESS",
+    "43": "I-API_KEY",
+    "44": "I-BANK_ROUTING_NUMBER",
+    "45": "I-BIOMETRIC_IDENTIFIER",
+    "46": "I-CERTIFICATE_LICENSE_NUMBER",
+    "47": "I-CITY",
+    "48": "I-COMPANY_NAME",
+    "49": "I-COORDINATE",
+    "50": "I-COUNTRY",
+    "51": "I-CREDIT_CARD_NUMBER",
+    "52": "I-CUSTOMER_ID",
+    "53": "I-CVV",
+    "54": "I-DATE",
+    "55": "I-DATE_OF_BIRTH",
+    "56": "I-DATE_TIME",
+    "57": "I-DEVICE_IDENTIFIER",
+    "58": "I-EMAIL",
+    "59": "I-EMPLOYEE_ID",
+    "60": "I-FIRST_NAME",
+    "61": "I-HEALTH_PLAN_BENEFICIARY_NUMBER",
+    "62": "I-IPADDRESS",
+    "63": "I-LAST_NAME",
+    "64": "I-LICENSE_PLATE",
+    "65": "I-MEDICAL_RECORD_NUMBER",
+    "66": "I-NAME",
+    "67": "I-NATIONAL_ID",
+    "68": "I-PASSWORD",
+    "69": "I-PHONE_NUMBER",
+    "70": "I-PIN",
+    "71": "I-POSTCODE",
+    "72": "I-SSN/CCCD",
+    "73": "I-STATE",
+    "74": "I-STREET_ADDRESS",
+    "75": "I-SWIFT_BIC",
+    "76": "I-TAX_ID",
+    "77": "I-TIME",
+    "78": "I-UNIQUE_IDENTIFIER",
+    "79": "I-URL",
+    "80": "I-USER_NAME",
+    "81": "I-VEHICLE_IDENTIFIER",
+    "82": "O"
+  },
+  "initializer_range": 0.02,
+  "intermediate_size": 3072,
+  "label2id": {
+    "B-ACCOUNT_NUMBER": 0,
+    "B-ADDRESS": 1,
+    "B-API_KEY": 2,
+    "B-BANK_ROUTING_NUMBER": 3,
+    "B-BIOMETRIC_IDENTIFIER": 4,
+    "B-CERTIFICATE_LICENSE_NUMBER": 5,
+    "B-CITY": 6,
+    "B-COMPANY_NAME": 7,
+    "B-COORDINATE": 8,
+    "B-COUNTRY": 9,
+    "B-CREDIT_CARD_NUMBER": 10,
+    "B-CUSTOMER_ID": 11,
+    "B-CVV": 12,
+    "B-DATE": 13,
+    "B-DATE_OF_BIRTH": 14,
+    "B-DATE_TIME": 15,
+    "B-DEVICE_IDENTIFIER": 16,
+    "B-EMAIL": 17,
+    "B-EMPLOYEE_ID": 18,
+    "B-FIRST_NAME": 19,
+    "B-HEALTH_PLAN_BENEFICIARY_NUMBER": 20,
+    "B-IPADDRESS": 21,
+    "B-LAST_NAME": 22,
+    "B-LICENSE_PLATE": 23,
+    "B-MEDICAL_RECORD_NUMBER": 24,
+    "B-NAME": 25,
+    "B-NATIONAL_ID": 26,
+    "B-PASSWORD": 27,
+    "B-PHONE_NUMBER": 28,
+    "B-PIN": 29,
+    "B-POSTCODE": 30,
+    "B-SSN/CCCD": 31,
+    "B-STATE": 32,
+    "B-STREET_ADDRESS": 33,
+    "B-SWIFT_BIC": 34,
+    "B-TAX_ID": 35,
+    "B-TIME": 36,
+    "B-UNIQUE_IDENTIFIER": 37,
+    "B-URL": 38,
+    "B-USER_NAME": 39,
+    "B-VEHICLE_IDENTIFIER": 40,
+    "I-ACCOUNT_NUMBER": 41,
+    "I-ADDRESS": 42,
+    "I-API_KEY": 43,
+    "I-BANK_ROUTING_NUMBER": 44,
+    "I-BIOMETRIC_IDENTIFIER": 45,
+    "I-CERTIFICATE_LICENSE_NUMBER": 46,
+    "I-CITY": 47,
+    "I-COMPANY_NAME": 48,
+    "I-COORDINATE": 49,
+    "I-COUNTRY": 50,
+    "I-CREDIT_CARD_NUMBER": 51,
+    "I-CUSTOMER_ID": 52,
+    "I-CVV": 53,
+    "I-DATE": 54,
+    "I-DATE_OF_BIRTH": 55,
+    "I-DATE_TIME": 56,
+    "I-DEVICE_IDENTIFIER": 57,
+    "I-EMAIL": 58,
+    "I-EMPLOYEE_ID": 59,
+    "I-FIRST_NAME": 60,
+    "I-HEALTH_PLAN_BENEFICIARY_NUMBER": 61,
+    "I-IPADDRESS": 62,
+    "I-LAST_NAME": 63,
+    "I-LICENSE_PLATE": 64,
+    "I-MEDICAL_RECORD_NUMBER": 65,
+    "I-NAME": 66,
+    "I-NATIONAL_ID": 67,
+    "I-PASSWORD": 68,
+    "I-PHONE_NUMBER": 69,
+    "I-PIN": 70,
+    "I-POSTCODE": 71,
+    "I-SSN/CCCD": 72,
+    "I-STATE": 73,
+    "I-STREET_ADDRESS": 74,
+    "I-SWIFT_BIC": 75,
+    "I-TAX_ID": 76,
+    "I-TIME": 77,
+    "I-UNIQUE_IDENTIFIER": 78,
+    "I-URL": 79,
+    "I-USER_NAME": 80,
+    "I-VEHICLE_IDENTIFIER": 81,
+    "O": 82
+  },
+  "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
+}

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

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:615a01d8f5a1ccbf233fe6e2effb060899e14af8433b20baf9bde3b088793778
+oid sha256:9373a4d407993ae77d6c24a301f5a8dccc7ee3e7fec365d4e7f282b4f8b9d55a
 size 711721104

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

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:0408505de4e49c1e6352ad3bfc9e4bf6cdaadae5346c991a6e07d0054acf1d80
+oid sha256:9f3fca44ea2c3269af512f4e8fcac91564061df8d0a811814cfddb25dfea554b
 size 5777