Hugging Face's logo

PuxAI
/
GenPII-framework

Safetensors
Model card Files
xet
 Community
PuxAI commited on
Commit
52f5d70
·
verified ·
1 Parent(s): 2f41b62

Upload folder using huggingface_hub

Browse files
Files changed (32) hide show
  1. gretel-pii-ready-augmented/TokenBased-BERT/checkpoint-1407/config.json +208 -0
  2. gretel-pii-ready-augmented/TokenBased-BERT/checkpoint-1407/model.safetensors +3 -0
  3. gretel-pii-ready-augmented/TokenBased-BERT/checkpoint-1407/optimizer.pt +3 -0
  4. gretel-pii-ready-augmented/TokenBased-BERT/checkpoint-1407/rng_state.pth +3 -0
  5. gretel-pii-ready-augmented/TokenBased-BERT/checkpoint-1407/scheduler.pt +3 -0
  6. gretel-pii-ready-augmented/TokenBased-BERT/checkpoint-1407/tokenizer.json +0 -0
  7. gretel-pii-ready-augmented/TokenBased-BERT/checkpoint-1407/tokenizer_config.json +15 -0
  8. gretel-pii-ready-augmented/TokenBased-BERT/checkpoint-1407/trainer_state.json +56 -0
  9. gretel-pii-ready-augmented/TokenBased-BERT/checkpoint-1407/training_args.bin +3 -0
  10. gretel-pii-ready-augmented/TokenBased-BERT/checkpoint-2814/config.json +208 -0
  11. gretel-pii-ready-augmented/TokenBased-BERT/checkpoint-2814/model.safetensors +3 -0
  12. gretel-pii-ready-augmented/TokenBased-BERT/checkpoint-2814/optimizer.pt +3 -0
  13. gretel-pii-ready-augmented/TokenBased-BERT/checkpoint-2814/rng_state.pth +3 -0
  14. gretel-pii-ready-augmented/TokenBased-BERT/checkpoint-2814/scheduler.pt +3 -0
  15. gretel-pii-ready-augmented/TokenBased-BERT/checkpoint-2814/tokenizer.json +0 -0
  16. gretel-pii-ready-augmented/TokenBased-BERT/checkpoint-2814/tokenizer_config.json +15 -0
  17. gretel-pii-ready-augmented/TokenBased-BERT/checkpoint-2814/trainer_state.json +85 -0
  18. gretel-pii-ready-augmented/TokenBased-BERT/checkpoint-2814/training_args.bin +3 -0
  19. gretel-pii-ready-augmented/TokenBased-BERT/checkpoint-4221/config.json +208 -0
  20. gretel-pii-ready-augmented/TokenBased-BERT/checkpoint-4221/model.safetensors +3 -0
  21. gretel-pii-ready-augmented/TokenBased-BERT/checkpoint-4221/optimizer.pt +3 -0
  22. gretel-pii-ready-augmented/TokenBased-BERT/checkpoint-4221/rng_state.pth +3 -0
  23. gretel-pii-ready-augmented/TokenBased-BERT/checkpoint-4221/scheduler.pt +3 -0
  24. gretel-pii-ready-augmented/TokenBased-BERT/checkpoint-4221/tokenizer.json +0 -0
  25. gretel-pii-ready-augmented/TokenBased-BERT/checkpoint-4221/tokenizer_config.json +15 -0
  26. gretel-pii-ready-augmented/TokenBased-BERT/checkpoint-4221/trainer_state.json +114 -0
  27. gretel-pii-ready-augmented/TokenBased-BERT/checkpoint-4221/training_args.bin +3 -0
  28. gretel-pii-ready-augmented/TokenBased-BERT/config.json +208 -0
  29. gretel-pii-ready-augmented/TokenBased-BERT/model.safetensors +3 -0
  30. gretel-pii-ready-augmented/TokenBased-BERT/tokenizer.json +0 -0
  31. gretel-pii-ready-augmented/TokenBased-BERT/tokenizer_config.json +15 -0
  32. gretel-pii-ready-augmented/TokenBased-BERT/training_args.bin +3 -0
gretel-pii-ready-augmented/TokenBased-BERT/checkpoint-1407/config.json ADDED
@@ -0,0 +1,208 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "add_cross_attention": false,
3
+ "architectures": [
4
+ "BertForTokenClassification"
5
+ ],
6
+ "attention_probs_dropout_prob": 0.1,
7
+ "bos_token_id": null,
8
+ "classifier_dropout": null,
9
+ "directionality": "bidi",
10
+ "dtype": "float32",
11
+ "eos_token_id": null,
12
+ "hidden_act": "gelu",
13
+ "hidden_dropout_prob": 0.1,
14
+ "hidden_size": 768,
15
+ "id2label": {
16
+ "0": "O",
17
+ "1": "B-ACCOUNT_NUMBER",
18
+ "2": "I-ACCOUNT_NUMBER",
19
+ "3": "B-ADDRESS",
20
+ "4": "I-ADDRESS",
21
+ "5": "B-API_KEY",
22
+ "6": "I-API_KEY",
23
+ "7": "B-BANK_ROUTING_NUMBER",
24
+ "8": "I-BANK_ROUTING_NUMBER",
25
+ "9": "B-BIOMETRIC_IDENTIFIER",
26
+ "10": "I-BIOMETRIC_IDENTIFIER",
27
+ "11": "B-CERTIFICATE_LICENSE_NUMBER",
28
+ "12": "I-CERTIFICATE_LICENSE_NUMBER",
29
+ "13": "B-CITY",
30
+ "14": "I-CITY",
31
+ "15": "B-COMPANY_NAME",
32
+ "16": "I-COMPANY_NAME",
33
+ "17": "B-COORDINATE",
34
+ "18": "I-COORDINATE",
35
+ "19": "B-COUNTRY",
36
+ "20": "I-COUNTRY",
37
+ "21": "B-CREDIT_CARD_NUMBER",
38
+ "22": "I-CREDIT_CARD_NUMBER",
39
+ "23": "B-CUSTOMER_ID",
40
+ "24": "I-CUSTOMER_ID",
41
+ "25": "B-CVV",
42
+ "26": "I-CVV",
43
+ "27": "B-DATE",
44
+ "28": "I-DATE",
45
+ "29": "B-DATE_OF_BIRTH",
46
+ "30": "I-DATE_OF_BIRTH",
47
+ "31": "B-DATE_TIME",
48
+ "32": "I-DATE_TIME",
49
+ "33": "B-DEVICE_IDENTIFIER",
50
+ "34": "I-DEVICE_IDENTIFIER",
51
+ "35": "B-EMAIL",
52
+ "36": "I-EMAIL",
53
+ "37": "B-EMPLOYEE_ID",
54
+ "38": "I-EMPLOYEE_ID",
55
+ "39": "B-FIRST_NAME",
56
+ "40": "I-FIRST_NAME",
57
+ "41": "B-HEALTH_PLAN_BENEFICIARY_NUMBER",
58
+ "42": "I-HEALTH_PLAN_BENEFICIARY_NUMBER",
59
+ "43": "B-IPV4",
60
+ "44": "I-IPV4",
61
+ "45": "B-IPV6",
62
+ "46": "I-IPV6",
63
+ "47": "B-LAST_NAME",
64
+ "48": "I-LAST_NAME",
65
+ "49": "B-LICENSE_PLATE",
66
+ "50": "I-LICENSE_PLATE",
67
+ "51": "B-MEDICAL_RECORD_NUMBER",
68
+ "52": "I-MEDICAL_RECORD_NUMBER",
69
+ "53": "B-NAME",
70
+ "54": "I-NAME",
71
+ "55": "B-NATIONAL_ID",
72
+ "56": "I-NATIONAL_ID",
73
+ "57": "B-PASSWORD",
74
+ "58": "I-PASSWORD",
75
+ "59": "B-PHONE_NUMBER",
76
+ "60": "I-PHONE_NUMBER",
77
+ "61": "B-PIN",
78
+ "62": "I-PIN",
79
+ "63": "B-POSTCODE",
80
+ "64": "I-POSTCODE",
81
+ "65": "B-SSN",
82
+ "66": "I-SSN",
83
+ "67": "B-STATE",
84
+ "68": "I-STATE",
85
+ "69": "B-STREET_ADDRESS",
86
+ "70": "I-STREET_ADDRESS",
87
+ "71": "B-SWIFT_BIC",
88
+ "72": "I-SWIFT_BIC",
89
+ "73": "B-TAX_ID",
90
+ "74": "I-TAX_ID",
91
+ "75": "B-TIME",
92
+ "76": "I-TIME",
93
+ "77": "B-UNIQUE_IDENTIFIER",
94
+ "78": "I-UNIQUE_IDENTIFIER",
95
+ "79": "B-URL",
96
+ "80": "I-URL",
97
+ "81": "B-USER_NAME",
98
+ "82": "I-USER_NAME",
99
+ "83": "B-VEHICLE_IDENTIFIER",
100
+ "84": "I-VEHICLE_IDENTIFIER"
101
+ },
102
+ "initializer_range": 0.02,
103
+ "intermediate_size": 3072,
104
+ "is_decoder": false,
105
+ "label2id": {
106
+ "B-ACCOUNT_NUMBER": 1,
107
+ "B-ADDRESS": 3,
108
+ "B-API_KEY": 5,
109
+ "B-BANK_ROUTING_NUMBER": 7,
110
+ "B-BIOMETRIC_IDENTIFIER": 9,
111
+ "B-CERTIFICATE_LICENSE_NUMBER": 11,
112
+ "B-CITY": 13,
113
+ "B-COMPANY_NAME": 15,
114
+ "B-COORDINATE": 17,
115
+ "B-COUNTRY": 19,
116
+ "B-CREDIT_CARD_NUMBER": 21,
117
+ "B-CUSTOMER_ID": 23,
118
+ "B-CVV": 25,
119
+ "B-DATE": 27,
120
+ "B-DATE_OF_BIRTH": 29,
121
+ "B-DATE_TIME": 31,
122
+ "B-DEVICE_IDENTIFIER": 33,
123
+ "B-EMAIL": 35,
124
+ "B-EMPLOYEE_ID": 37,
125
+ "B-FIRST_NAME": 39,
126
+ "B-HEALTH_PLAN_BENEFICIARY_NUMBER": 41,
127
+ "B-IPV4": 43,
128
+ "B-IPV6": 45,
129
+ "B-LAST_NAME": 47,
130
+ "B-LICENSE_PLATE": 49,
131
+ "B-MEDICAL_RECORD_NUMBER": 51,
132
+ "B-NAME": 53,
133
+ "B-NATIONAL_ID": 55,
134
+ "B-PASSWORD": 57,
135
+ "B-PHONE_NUMBER": 59,
136
+ "B-PIN": 61,
137
+ "B-POSTCODE": 63,
138
+ "B-SSN": 65,
139
+ "B-STATE": 67,
140
+ "B-STREET_ADDRESS": 69,
141
+ "B-SWIFT_BIC": 71,
142
+ "B-TAX_ID": 73,
143
+ "B-TIME": 75,
144
+ "B-UNIQUE_IDENTIFIER": 77,
145
+ "B-URL": 79,
146
+ "B-USER_NAME": 81,
147
+ "B-VEHICLE_IDENTIFIER": 83,
148
+ "I-ACCOUNT_NUMBER": 2,
149
+ "I-ADDRESS": 4,
150
+ "I-API_KEY": 6,
151
+ "I-BANK_ROUTING_NUMBER": 8,
152
+ "I-BIOMETRIC_IDENTIFIER": 10,
153
+ "I-CERTIFICATE_LICENSE_NUMBER": 12,
154
+ "I-CITY": 14,
155
+ "I-COMPANY_NAME": 16,
156
+ "I-COORDINATE": 18,
157
+ "I-COUNTRY": 20,
158
+ "I-CREDIT_CARD_NUMBER": 22,
159
+ "I-CUSTOMER_ID": 24,
160
+ "I-CVV": 26,
161
+ "I-DATE": 28,
162
+ "I-DATE_OF_BIRTH": 30,
163
+ "I-DATE_TIME": 32,
164
+ "I-DEVICE_IDENTIFIER": 34,
165
+ "I-EMAIL": 36,
166
+ "I-EMPLOYEE_ID": 38,
167
+ "I-FIRST_NAME": 40,
168
+ "I-HEALTH_PLAN_BENEFICIARY_NUMBER": 42,
169
+ "I-IPV4": 44,
170
+ "I-IPV6": 46,
171
+ "I-LAST_NAME": 48,
172
+ "I-LICENSE_PLATE": 50,
173
+ "I-MEDICAL_RECORD_NUMBER": 52,
174
+ "I-NAME": 54,
175
+ "I-NATIONAL_ID": 56,
176
+ "I-PASSWORD": 58,
177
+ "I-PHONE_NUMBER": 60,
178
+ "I-PIN": 62,
179
+ "I-POSTCODE": 64,
180
+ "I-SSN": 66,
181
+ "I-STATE": 68,
182
+ "I-STREET_ADDRESS": 70,
183
+ "I-SWIFT_BIC": 72,
184
+ "I-TAX_ID": 74,
185
+ "I-TIME": 76,
186
+ "I-UNIQUE_IDENTIFIER": 78,
187
+ "I-URL": 80,
188
+ "I-USER_NAME": 82,
189
+ "I-VEHICLE_IDENTIFIER": 84,
190
+ "O": 0
191
+ },
192
+ "layer_norm_eps": 1e-12,
193
+ "max_position_embeddings": 512,
194
+ "model_type": "bert",
195
+ "num_attention_heads": 12,
196
+ "num_hidden_layers": 12,
197
+ "pad_token_id": 0,
198
+ "pooler_fc_size": 768,
199
+ "pooler_num_attention_heads": 12,
200
+ "pooler_num_fc_layers": 3,
201
+ "pooler_size_per_head": 128,
202
+ "pooler_type": "first_token_transform",
203
+ "tie_word_embeddings": true,
204
+ "transformers_version": "5.3.0",
205
+ "type_vocab_size": 2,
206
+ "use_cache": false,
207
+ "vocab_size": 119547
208
+ }
gretel-pii-ready-augmented/TokenBased-BERT/checkpoint-1407/model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f2afbbaf6782ee18f1a6c7c7bb90703b0ee8121981eb39c0b72c468119e4dede
3
+ size 709336188
gretel-pii-ready-augmented/TokenBased-BERT/checkpoint-1407/optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:349dec90f8441a5c0ac21b3b749f7fd50e6e57a5eb84343f3231e99267403792
3
+ size 1418795403
gretel-pii-ready-augmented/TokenBased-BERT/checkpoint-1407/rng_state.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:42f46c372aa8862be7ba5526334ff4e288ccfb911c74cf171103225f029f7ae8
3
+ size 14645
gretel-pii-ready-augmented/TokenBased-BERT/checkpoint-1407/scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c6715a5a1995d6ff0489ea646102a508f7923079efd09ce587ae600a7314d1eb
3
+ size 1465
gretel-pii-ready-augmented/TokenBased-BERT/checkpoint-1407/tokenizer.json ADDED
The diff for this file is too large to render. See raw diff
 
gretel-pii-ready-augmented/TokenBased-BERT/checkpoint-1407/tokenizer_config.json ADDED
@@ -0,0 +1,15 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "add_prefix_space": true,
3
+ "backend": "tokenizers",
4
+ "cls_token": "[CLS]",
5
+ "do_lower_case": false,
6
+ "is_local": false,
7
+ "mask_token": "[MASK]",
8
+ "model_max_length": 512,
9
+ "pad_token": "[PAD]",
10
+ "sep_token": "[SEP]",
11
+ "strip_accents": null,
12
+ "tokenize_chinese_chars": true,
13
+ "tokenizer_class": "BertTokenizer",
14
+ "unk_token": "[UNK]"
15
+ }
gretel-pii-ready-augmented/TokenBased-BERT/checkpoint-1407/trainer_state.json ADDED
@@ -0,0 +1,56 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_global_step": null,
3
+ "best_metric": null,
4
+ "best_model_checkpoint": null,
5
+ "epoch": 1.0,
6
+ "eval_steps": 500,
7
+ "global_step": 1407,
8
+ "is_hyper_param_search": false,
9
+ "is_local_process_zero": true,
10
+ "is_world_process_zero": true,
11
+ "log_history": [
12
+ {
13
+ "epoch": 0.35536602700781805,
14
+ "grad_norm": 1.2672953605651855,
15
+ "learning_rate": 1.7635631366974652e-05,
16
+ "loss": 0.342183837890625,
17
+ "step": 500
18
+ },
19
+ {
20
+ "epoch": 0.7107320540156361,
21
+ "grad_norm": 1.0100759267807007,
22
+ "learning_rate": 1.5266524520255865e-05,
23
+ "loss": 0.082106201171875,
24
+ "step": 1000
25
+ },
26
+ {
27
+ "epoch": 1.0,
28
+ "eval_loss": 0.05113217607140541,
29
+ "eval_runtime": 18.3397,
30
+ "eval_samples_per_second": 545.265,
31
+ "eval_steps_per_second": 68.158,
32
+ "step": 1407
33
+ }
34
+ ],
35
+ "logging_steps": 500,
36
+ "max_steps": 4221,
37
+ "num_input_tokens_seen": 0,
38
+ "num_train_epochs": 3,
39
+ "save_steps": 500,
40
+ "stateful_callbacks": {
41
+ "TrainerControl": {
42
+ "args": {
43
+ "should_epoch_stop": false,
44
+ "should_evaluate": false,
45
+ "should_log": false,
46
+ "should_save": true,
47
+ "should_training_stop": false
48
+ },
49
+ "attributes": {}
50
+ }
51
+ },
52
+ "total_flos": 2.35343549952e+16,
53
+ "train_batch_size": 64,
54
+ "trial_name": null,
55
+ "trial_params": null
56
+ }
gretel-pii-ready-augmented/TokenBased-BERT/checkpoint-1407/training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:54c2c0c4bc66849033182b83cc6ed373239dd57860c0cb28f90d25bee0ef7ded
3
+ size 5201
gretel-pii-ready-augmented/TokenBased-BERT/checkpoint-2814/config.json ADDED
@@ -0,0 +1,208 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "add_cross_attention": false,
3
+ "architectures": [
4
+ "BertForTokenClassification"
5
+ ],
6
+ "attention_probs_dropout_prob": 0.1,
7
+ "bos_token_id": null,
8
+ "classifier_dropout": null,
9
+ "directionality": "bidi",
10
+ "dtype": "float32",
11
+ "eos_token_id": null,
12
+ "hidden_act": "gelu",
13
+ "hidden_dropout_prob": 0.1,
14
+ "hidden_size": 768,
15
+ "id2label": {
16
+ "0": "O",
17
+ "1": "B-ACCOUNT_NUMBER",
18
+ "2": "I-ACCOUNT_NUMBER",
19
+ "3": "B-ADDRESS",
20
+ "4": "I-ADDRESS",
21
+ "5": "B-API_KEY",
22
+ "6": "I-API_KEY",
23
+ "7": "B-BANK_ROUTING_NUMBER",
24
+ "8": "I-BANK_ROUTING_NUMBER",
25
+ "9": "B-BIOMETRIC_IDENTIFIER",
26
+ "10": "I-BIOMETRIC_IDENTIFIER",
27
+ "11": "B-CERTIFICATE_LICENSE_NUMBER",
28
+ "12": "I-CERTIFICATE_LICENSE_NUMBER",
29
+ "13": "B-CITY",
30
+ "14": "I-CITY",
31
+ "15": "B-COMPANY_NAME",
32
+ "16": "I-COMPANY_NAME",
33
+ "17": "B-COORDINATE",
34
+ "18": "I-COORDINATE",
35
+ "19": "B-COUNTRY",
36
+ "20": "I-COUNTRY",
37
+ "21": "B-CREDIT_CARD_NUMBER",
38
+ "22": "I-CREDIT_CARD_NUMBER",
39
+ "23": "B-CUSTOMER_ID",
40
+ "24": "I-CUSTOMER_ID",
41
+ "25": "B-CVV",
42
+ "26": "I-CVV",
43
+ "27": "B-DATE",
44
+ "28": "I-DATE",
45
+ "29": "B-DATE_OF_BIRTH",
46
+ "30": "I-DATE_OF_BIRTH",
47
+ "31": "B-DATE_TIME",
48
+ "32": "I-DATE_TIME",
49
+ "33": "B-DEVICE_IDENTIFIER",
50
+ "34": "I-DEVICE_IDENTIFIER",
51
+ "35": "B-EMAIL",
52
+ "36": "I-EMAIL",
53
+ "37": "B-EMPLOYEE_ID",
54
+ "38": "I-EMPLOYEE_ID",
55
+ "39": "B-FIRST_NAME",
56
+ "40": "I-FIRST_NAME",
57
+ "41": "B-HEALTH_PLAN_BENEFICIARY_NUMBER",
58
+ "42": "I-HEALTH_PLAN_BENEFICIARY_NUMBER",
59
+ "43": "B-IPV4",
60
+ "44": "I-IPV4",
61
+ "45": "B-IPV6",
62
+ "46": "I-IPV6",
63
+ "47": "B-LAST_NAME",
64
+ "48": "I-LAST_NAME",
65
+ "49": "B-LICENSE_PLATE",
66
+ "50": "I-LICENSE_PLATE",
67
+ "51": "B-MEDICAL_RECORD_NUMBER",
68
+ "52": "I-MEDICAL_RECORD_NUMBER",
69
+ "53": "B-NAME",
70
+ "54": "I-NAME",
71
+ "55": "B-NATIONAL_ID",
72
+ "56": "I-NATIONAL_ID",
73
+ "57": "B-PASSWORD",
74
+ "58": "I-PASSWORD",
75
+ "59": "B-PHONE_NUMBER",
76
+ "60": "I-PHONE_NUMBER",
77
+ "61": "B-PIN",
78
+ "62": "I-PIN",
79
+ "63": "B-POSTCODE",
80
+ "64": "I-POSTCODE",
81
+ "65": "B-SSN",
82
+ "66": "I-SSN",
83
+ "67": "B-STATE",
84
+ "68": "I-STATE",
85
+ "69": "B-STREET_ADDRESS",
86
+ "70": "I-STREET_ADDRESS",
87
+ "71": "B-SWIFT_BIC",
88
+ "72": "I-SWIFT_BIC",
89
+ "73": "B-TAX_ID",
90
+ "74": "I-TAX_ID",
91
+ "75": "B-TIME",
92
+ "76": "I-TIME",
93
+ "77": "B-UNIQUE_IDENTIFIER",
94
+ "78": "I-UNIQUE_IDENTIFIER",
95
+ "79": "B-URL",
96
+ "80": "I-URL",
97
+ "81": "B-USER_NAME",
98
+ "82": "I-USER_NAME",
99
+ "83": "B-VEHICLE_IDENTIFIER",
100
+ "84": "I-VEHICLE_IDENTIFIER"
101
+ },
102
+ "initializer_range": 0.02,
103
+ "intermediate_size": 3072,
104
+ "is_decoder": false,
105
+ "label2id": {
106
+ "B-ACCOUNT_NUMBER": 1,
107
+ "B-ADDRESS": 3,
108
+ "B-API_KEY": 5,
109
+ "B-BANK_ROUTING_NUMBER": 7,
110
+ "B-BIOMETRIC_IDENTIFIER": 9,
111
+ "B-CERTIFICATE_LICENSE_NUMBER": 11,
112
+ "B-CITY": 13,
113
+ "B-COMPANY_NAME": 15,
114
+ "B-COORDINATE": 17,
115
+ "B-COUNTRY": 19,
116
+ "B-CREDIT_CARD_NUMBER": 21,
117
+ "B-CUSTOMER_ID": 23,
118
+ "B-CVV": 25,
119
+ "B-DATE": 27,
120
+ "B-DATE_OF_BIRTH": 29,
121
+ "B-DATE_TIME": 31,
122
+ "B-DEVICE_IDENTIFIER": 33,
123
+ "B-EMAIL": 35,
124
+ "B-EMPLOYEE_ID": 37,
125
+ "B-FIRST_NAME": 39,
126
+ "B-HEALTH_PLAN_BENEFICIARY_NUMBER": 41,
127
+ "B-IPV4": 43,
128
+ "B-IPV6": 45,
129
+ "B-LAST_NAME": 47,
130
+ "B-LICENSE_PLATE": 49,
131
+ "B-MEDICAL_RECORD_NUMBER": 51,
132
+ "B-NAME": 53,
133
+ "B-NATIONAL_ID": 55,
134
+ "B-PASSWORD": 57,
135
+ "B-PHONE_NUMBER": 59,
136
+ "B-PIN": 61,
137
+ "B-POSTCODE": 63,
138
+ "B-SSN": 65,
139
+ "B-STATE": 67,
140
+ "B-STREET_ADDRESS": 69,
141
+ "B-SWIFT_BIC": 71,
142
+ "B-TAX_ID": 73,
143
+ "B-TIME": 75,
144
+ "B-UNIQUE_IDENTIFIER": 77,
145
+ "B-URL": 79,
146
+ "B-USER_NAME": 81,
147
+ "B-VEHICLE_IDENTIFIER": 83,
148
+ "I-ACCOUNT_NUMBER": 2,
149
+ "I-ADDRESS": 4,
150
+ "I-API_KEY": 6,
151
+ "I-BANK_ROUTING_NUMBER": 8,
152
+ "I-BIOMETRIC_IDENTIFIER": 10,
153
+ "I-CERTIFICATE_LICENSE_NUMBER": 12,
154
+ "I-CITY": 14,
155
+ "I-COMPANY_NAME": 16,
156
+ "I-COORDINATE": 18,
157
+ "I-COUNTRY": 20,
158
+ "I-CREDIT_CARD_NUMBER": 22,
159
+ "I-CUSTOMER_ID": 24,
160
+ "I-CVV": 26,
161
+ "I-DATE": 28,
162
+ "I-DATE_OF_BIRTH": 30,
163
+ "I-DATE_TIME": 32,
164
+ "I-DEVICE_IDENTIFIER": 34,
165
+ "I-EMAIL": 36,
166
+ "I-EMPLOYEE_ID": 38,
167
+ "I-FIRST_NAME": 40,
168
+ "I-HEALTH_PLAN_BENEFICIARY_NUMBER": 42,
169
+ "I-IPV4": 44,
170
+ "I-IPV6": 46,
171
+ "I-LAST_NAME": 48,
172
+ "I-LICENSE_PLATE": 50,
173
+ "I-MEDICAL_RECORD_NUMBER": 52,
174
+ "I-NAME": 54,
175
+ "I-NATIONAL_ID": 56,
176
+ "I-PASSWORD": 58,
177
+ "I-PHONE_NUMBER": 60,
178
+ "I-PIN": 62,
179
+ "I-POSTCODE": 64,
180
+ "I-SSN": 66,
181
+ "I-STATE": 68,
182
+ "I-STREET_ADDRESS": 70,
183
+ "I-SWIFT_BIC": 72,
184
+ "I-TAX_ID": 74,
185
+ "I-TIME": 76,
186
+ "I-UNIQUE_IDENTIFIER": 78,
187
+ "I-URL": 80,
188
+ "I-USER_NAME": 82,
189
+ "I-VEHICLE_IDENTIFIER": 84,
190
+ "O": 0
191
+ },
192
+ "layer_norm_eps": 1e-12,
193
+ "max_position_embeddings": 512,
194
+ "model_type": "bert",
195
+ "num_attention_heads": 12,
196
+ "num_hidden_layers": 12,
197
+ "pad_token_id": 0,
198
+ "pooler_fc_size": 768,
199
+ "pooler_num_attention_heads": 12,
200
+ "pooler_num_fc_layers": 3,
201
+ "pooler_size_per_head": 128,
202
+ "pooler_type": "first_token_transform",
203
+ "tie_word_embeddings": true,
204
+ "transformers_version": "5.3.0",
205
+ "type_vocab_size": 2,
206
+ "use_cache": false,
207
+ "vocab_size": 119547
208
+ }
gretel-pii-ready-augmented/TokenBased-BERT/checkpoint-2814/model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3a215e09bc553e3405a88429c81eb5da923944db909e49c5adcbf09107a52408
3
+ size 709336188
gretel-pii-ready-augmented/TokenBased-BERT/checkpoint-2814/optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a23eb2a4f5b32ee39ca950d73a6734dae3f48740fdc853bdb6d30c57e011a5dc
3
+ size 1418795403
gretel-pii-ready-augmented/TokenBased-BERT/checkpoint-2814/rng_state.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:946e3e88595957f9008ab6278a369044f36801cee1b0e02f41fd1c8fe859a88e
3
+ size 14645
gretel-pii-ready-augmented/TokenBased-BERT/checkpoint-2814/scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:14092e99b0f802054b8990cd9c06f451e94a8d55bcfbb236456db40050b79d23
3
+ size 1465
gretel-pii-ready-augmented/TokenBased-BERT/checkpoint-2814/tokenizer.json ADDED
The diff for this file is too large to render. See raw diff
 
gretel-pii-ready-augmented/TokenBased-BERT/checkpoint-2814/tokenizer_config.json ADDED
@@ -0,0 +1,15 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "add_prefix_space": true,
3
+ "backend": "tokenizers",
4
+ "cls_token": "[CLS]",
5
+ "do_lower_case": false,
6
+ "is_local": false,
7
+ "mask_token": "[MASK]",
8
+ "model_max_length": 512,
9
+ "pad_token": "[PAD]",
10
+ "sep_token": "[SEP]",
11
+ "strip_accents": null,
12
+ "tokenize_chinese_chars": true,
13
+ "tokenizer_class": "BertTokenizer",
14
+ "unk_token": "[UNK]"
15
+ }
gretel-pii-ready-augmented/TokenBased-BERT/checkpoint-2814/trainer_state.json ADDED
@@ -0,0 +1,85 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_global_step": null,
3
+ "best_metric": null,
4
+ "best_model_checkpoint": null,
5
+ "epoch": 2.0,
6
+ "eval_steps": 500,
7
+ "global_step": 2814,
8
+ "is_hyper_param_search": false,
9
+ "is_local_process_zero": true,
10
+ "is_world_process_zero": true,
11
+ "log_history": [
12
+ {
13
+ "epoch": 0.35536602700781805,
14
+ "grad_norm": 1.2672953605651855,
15
+ "learning_rate": 1.7635631366974652e-05,
16
+ "loss": 0.342183837890625,
17
+ "step": 500
18
+ },
19
+ {
20
+ "epoch": 0.7107320540156361,
21
+ "grad_norm": 1.0100759267807007,
22
+ "learning_rate": 1.5266524520255865e-05,
23
+ "loss": 0.082106201171875,
24
+ "step": 1000
25
+ },
26
+ {
27
+ "epoch": 1.0,
28
+ "eval_loss": 0.05113217607140541,
29
+ "eval_runtime": 18.3397,
30
+ "eval_samples_per_second": 545.265,
31
+ "eval_steps_per_second": 68.158,
32
+ "step": 1407
33
+ },
34
+ {
35
+ "epoch": 1.0660980810234542,
36
+ "grad_norm": 1.1233466863632202,
37
+ "learning_rate": 1.2897417673537079e-05,
38
+ "loss": 0.06107857894897461,
39
+ "step": 1500
40
+ },
41
+ {
42
+ "epoch": 1.4214641080312722,
43
+ "grad_norm": 0.954052746295929,
44
+ "learning_rate": 1.052831082681829e-05,
45
+ "loss": 0.05122759628295898,
46
+ "step": 2000
47
+ },
48
+ {
49
+ "epoch": 1.7768301350390903,
50
+ "grad_norm": 0.5427104234695435,
51
+ "learning_rate": 8.159203980099502e-06,
52
+ "loss": 0.0449744873046875,
53
+ "step": 2500
54
+ },
55
+ {
56
+ "epoch": 2.0,
57
+ "eval_loss": 0.041359808295965195,
58
+ "eval_runtime": 18.3676,
59
+ "eval_samples_per_second": 544.436,
60
+ "eval_steps_per_second": 68.055,
61
+ "step": 2814
62
+ }
63
+ ],
64
+ "logging_steps": 500,
65
+ "max_steps": 4221,
66
+ "num_input_tokens_seen": 0,
67
+ "num_train_epochs": 3,
68
+ "save_steps": 500,
69
+ "stateful_callbacks": {
70
+ "TrainerControl": {
71
+ "args": {
72
+ "should_epoch_stop": false,
73
+ "should_evaluate": false,
74
+ "should_log": false,
75
+ "should_save": true,
76
+ "should_training_stop": false
77
+ },
78
+ "attributes": {}
79
+ }
80
+ },
81
+ "total_flos": 4.70687099904e+16,
82
+ "train_batch_size": 64,
83
+ "trial_name": null,
84
+ "trial_params": null
85
+ }
gretel-pii-ready-augmented/TokenBased-BERT/checkpoint-2814/training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:54c2c0c4bc66849033182b83cc6ed373239dd57860c0cb28f90d25bee0ef7ded
3
+ size 5201
gretel-pii-ready-augmented/TokenBased-BERT/checkpoint-4221/config.json ADDED
@@ -0,0 +1,208 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "add_cross_attention": false,
3
+ "architectures": [
4
+ "BertForTokenClassification"
5
+ ],
6
+ "attention_probs_dropout_prob": 0.1,
7
+ "bos_token_id": null,
8
+ "classifier_dropout": null,
9
+ "directionality": "bidi",
10
+ "dtype": "float32",
11
+ "eos_token_id": null,
12
+ "hidden_act": "gelu",
13
+ "hidden_dropout_prob": 0.1,
14
+ "hidden_size": 768,
15
+ "id2label": {
16
+ "0": "O",
17
+ "1": "B-ACCOUNT_NUMBER",
18
+ "2": "I-ACCOUNT_NUMBER",
19
+ "3": "B-ADDRESS",
20
+ "4": "I-ADDRESS",
21
+ "5": "B-API_KEY",
22
+ "6": "I-API_KEY",
23
+ "7": "B-BANK_ROUTING_NUMBER",
24
+ "8": "I-BANK_ROUTING_NUMBER",
25
+ "9": "B-BIOMETRIC_IDENTIFIER",
26
+ "10": "I-BIOMETRIC_IDENTIFIER",
27
+ "11": "B-CERTIFICATE_LICENSE_NUMBER",
28
+ "12": "I-CERTIFICATE_LICENSE_NUMBER",
29
+ "13": "B-CITY",
30
+ "14": "I-CITY",
31
+ "15": "B-COMPANY_NAME",
32
+ "16": "I-COMPANY_NAME",
33
+ "17": "B-COORDINATE",
34
+ "18": "I-COORDINATE",
35
+ "19": "B-COUNTRY",
36
+ "20": "I-COUNTRY",
37
+ "21": "B-CREDIT_CARD_NUMBER",
38
+ "22": "I-CREDIT_CARD_NUMBER",
39
+ "23": "B-CUSTOMER_ID",
40
+ "24": "I-CUSTOMER_ID",
41
+ "25": "B-CVV",
42
+ "26": "I-CVV",
43
+ "27": "B-DATE",
44
+ "28": "I-DATE",
45
+ "29": "B-DATE_OF_BIRTH",
46
+ "30": "I-DATE_OF_BIRTH",
47
+ "31": "B-DATE_TIME",
48
+ "32": "I-DATE_TIME",
49
+ "33": "B-DEVICE_IDENTIFIER",
50
+ "34": "I-DEVICE_IDENTIFIER",
51
+ "35": "B-EMAIL",
52
+ "36": "I-EMAIL",
53
+ "37": "B-EMPLOYEE_ID",
54
+ "38": "I-EMPLOYEE_ID",
55
+ "39": "B-FIRST_NAME",
56
+ "40": "I-FIRST_NAME",
57
+ "41": "B-HEALTH_PLAN_BENEFICIARY_NUMBER",
58
+ "42": "I-HEALTH_PLAN_BENEFICIARY_NUMBER",
59
+ "43": "B-IPV4",
60
+ "44": "I-IPV4",
61
+ "45": "B-IPV6",
62
+ "46": "I-IPV6",
63
+ "47": "B-LAST_NAME",
64
+ "48": "I-LAST_NAME",
65
+ "49": "B-LICENSE_PLATE",
66
+ "50": "I-LICENSE_PLATE",
67
+ "51": "B-MEDICAL_RECORD_NUMBER",
68
+ "52": "I-MEDICAL_RECORD_NUMBER",
69
+ "53": "B-NAME",
70
+ "54": "I-NAME",
71
+ "55": "B-NATIONAL_ID",
72
+ "56": "I-NATIONAL_ID",
73
+ "57": "B-PASSWORD",
74
+ "58": "I-PASSWORD",
75
+ "59": "B-PHONE_NUMBER",
76
+ "60": "I-PHONE_NUMBER",
77
+ "61": "B-PIN",
78
+ "62": "I-PIN",
79
+ "63": "B-POSTCODE",
80
+ "64": "I-POSTCODE",
81
+ "65": "B-SSN",
82
+ "66": "I-SSN",
83
+ "67": "B-STATE",
84
+ "68": "I-STATE",
85
+ "69": "B-STREET_ADDRESS",
86
+ "70": "I-STREET_ADDRESS",
87
+ "71": "B-SWIFT_BIC",
88
+ "72": "I-SWIFT_BIC",
89
+ "73": "B-TAX_ID",
90
+ "74": "I-TAX_ID",
91
+ "75": "B-TIME",
92
+ "76": "I-TIME",
93
+ "77": "B-UNIQUE_IDENTIFIER",
94
+ "78": "I-UNIQUE_IDENTIFIER",
95
+ "79": "B-URL",
96
+ "80": "I-URL",
97
+ "81": "B-USER_NAME",
98
+ "82": "I-USER_NAME",
99
+ "83": "B-VEHICLE_IDENTIFIER",
100
+ "84": "I-VEHICLE_IDENTIFIER"
101
+ },
102
+ "initializer_range": 0.02,
103
+ "intermediate_size": 3072,
104
+ "is_decoder": false,
105
+ "label2id": {
106
+ "B-ACCOUNT_NUMBER": 1,
107
+ "B-ADDRESS": 3,
108
+ "B-API_KEY": 5,
109
+ "B-BANK_ROUTING_NUMBER": 7,
110
+ "B-BIOMETRIC_IDENTIFIER": 9,
111
+ "B-CERTIFICATE_LICENSE_NUMBER": 11,
112
+ "B-CITY": 13,
113
+ "B-COMPANY_NAME": 15,
114
+ "B-COORDINATE": 17,
115
+ "B-COUNTRY": 19,
116
+ "B-CREDIT_CARD_NUMBER": 21,
117
+ "B-CUSTOMER_ID": 23,
118
+ "B-CVV": 25,
119
+ "B-DATE": 27,
120
+ "B-DATE_OF_BIRTH": 29,
121
+ "B-DATE_TIME": 31,
122
+ "B-DEVICE_IDENTIFIER": 33,
123
+ "B-EMAIL": 35,
124
+ "B-EMPLOYEE_ID": 37,
125
+ "B-FIRST_NAME": 39,
126
+ "B-HEALTH_PLAN_BENEFICIARY_NUMBER": 41,
127
+ "B-IPV4": 43,
128
+ "B-IPV6": 45,
129
+ "B-LAST_NAME": 47,
130
+ "B-LICENSE_PLATE": 49,
131
+ "B-MEDICAL_RECORD_NUMBER": 51,
132
+ "B-NAME": 53,
133
+ "B-NATIONAL_ID": 55,
134
+ "B-PASSWORD": 57,
135
+ "B-PHONE_NUMBER": 59,
136
+ "B-PIN": 61,
137
+ "B-POSTCODE": 63,
138
+ "B-SSN": 65,
139
+ "B-STATE": 67,
140
+ "B-STREET_ADDRESS": 69,
141
+ "B-SWIFT_BIC": 71,
142
+ "B-TAX_ID": 73,
143
+ "B-TIME": 75,
144
+ "B-UNIQUE_IDENTIFIER": 77,
145
+ "B-URL": 79,
146
+ "B-USER_NAME": 81,
147
+ "B-VEHICLE_IDENTIFIER": 83,
148
+ "I-ACCOUNT_NUMBER": 2,
149
+ "I-ADDRESS": 4,
150
+ "I-API_KEY": 6,
151
+ "I-BANK_ROUTING_NUMBER": 8,
152
+ "I-BIOMETRIC_IDENTIFIER": 10,
153
+ "I-CERTIFICATE_LICENSE_NUMBER": 12,
154
+ "I-CITY": 14,
155
+ "I-COMPANY_NAME": 16,
156
+ "I-COORDINATE": 18,
157
+ "I-COUNTRY": 20,
158
+ "I-CREDIT_CARD_NUMBER": 22,
159
+ "I-CUSTOMER_ID": 24,
160
+ "I-CVV": 26,
161
+ "I-DATE": 28,
162
+ "I-DATE_OF_BIRTH": 30,
163
+ "I-DATE_TIME": 32,
164
+ "I-DEVICE_IDENTIFIER": 34,
165
+ "I-EMAIL": 36,
166
+ "I-EMPLOYEE_ID": 38,
167
+ "I-FIRST_NAME": 40,
168
+ "I-HEALTH_PLAN_BENEFICIARY_NUMBER": 42,
169
+ "I-IPV4": 44,
170
+ "I-IPV6": 46,
171
+ "I-LAST_NAME": 48,
172
+ "I-LICENSE_PLATE": 50,
173
+ "I-MEDICAL_RECORD_NUMBER": 52,
174
+ "I-NAME": 54,
175
+ "I-NATIONAL_ID": 56,
176
+ "I-PASSWORD": 58,
177
+ "I-PHONE_NUMBER": 60,
178
+ "I-PIN": 62,
179
+ "I-POSTCODE": 64,
180
+ "I-SSN": 66,
181
+ "I-STATE": 68,
182
+ "I-STREET_ADDRESS": 70,
183
+ "I-SWIFT_BIC": 72,
184
+ "I-TAX_ID": 74,
185
+ "I-TIME": 76,
186
+ "I-UNIQUE_IDENTIFIER": 78,
187
+ "I-URL": 80,
188
+ "I-USER_NAME": 82,
189
+ "I-VEHICLE_IDENTIFIER": 84,
190
+ "O": 0
191
+ },
192
+ "layer_norm_eps": 1e-12,
193
+ "max_position_embeddings": 512,
194
+ "model_type": "bert",
195
+ "num_attention_heads": 12,
196
+ "num_hidden_layers": 12,
197
+ "pad_token_id": 0,
198
+ "pooler_fc_size": 768,
199
+ "pooler_num_attention_heads": 12,
200
+ "pooler_num_fc_layers": 3,
201
+ "pooler_size_per_head": 128,
202
+ "pooler_type": "first_token_transform",
203
+ "tie_word_embeddings": true,
204
+ "transformers_version": "5.3.0",
205
+ "type_vocab_size": 2,
206
+ "use_cache": false,
207
+ "vocab_size": 119547
208
+ }
gretel-pii-ready-augmented/TokenBased-BERT/checkpoint-4221/model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f7c2d3ed068f24221ef2a2864932176a9fc54ebf7fa71b26bfa4ece39f72bcda
3
+ size 709336188
gretel-pii-ready-augmented/TokenBased-BERT/checkpoint-4221/optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:cbaf89faa868070fac1d5b76c714d76e91d57b16f5ffa68a011bcbb02b52fe98
3
+ size 1418795403
gretel-pii-ready-augmented/TokenBased-BERT/checkpoint-4221/rng_state.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7ae0cd324fcc63c40167d970de0fe8fcf2e939f6fbc417de9b909e897861982b
3
+ size 14645
gretel-pii-ready-augmented/TokenBased-BERT/checkpoint-4221/scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:69b52d799f27b7647d5ff11de1e48fc331b2d5d2a0dcf8482a15b15f3fd4ef4d
3
+ size 1465
gretel-pii-ready-augmented/TokenBased-BERT/checkpoint-4221/tokenizer.json ADDED
The diff for this file is too large to render. See raw diff
 
gretel-pii-ready-augmented/TokenBased-BERT/checkpoint-4221/tokenizer_config.json ADDED
@@ -0,0 +1,15 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "add_prefix_space": true,
3
+ "backend": "tokenizers",
4
+ "cls_token": "[CLS]",
5
+ "do_lower_case": false,
6
+ "is_local": false,
7
+ "mask_token": "[MASK]",
8
+ "model_max_length": 512,
9
+ "pad_token": "[PAD]",
10
+ "sep_token": "[SEP]",
11
+ "strip_accents": null,
12
+ "tokenize_chinese_chars": true,
13
+ "tokenizer_class": "BertTokenizer",
14
+ "unk_token": "[UNK]"
15
+ }
gretel-pii-ready-augmented/TokenBased-BERT/checkpoint-4221/trainer_state.json ADDED
@@ -0,0 +1,114 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_global_step": null,
3
+ "best_metric": null,
4
+ "best_model_checkpoint": null,
5
+ "epoch": 3.0,
6
+ "eval_steps": 500,
7
+ "global_step": 4221,
8
+ "is_hyper_param_search": false,
9
+ "is_local_process_zero": true,
10
+ "is_world_process_zero": true,
11
+ "log_history": [
12
+ {
13
+ "epoch": 0.35536602700781805,
14
+ "grad_norm": 1.2672953605651855,
15
+ "learning_rate": 1.7635631366974652e-05,
16
+ "loss": 0.342183837890625,
17
+ "step": 500
18
+ },
19
+ {
20
+ "epoch": 0.7107320540156361,
21
+ "grad_norm": 1.0100759267807007,
22
+ "learning_rate": 1.5266524520255865e-05,
23
+ "loss": 0.082106201171875,
24
+ "step": 1000
25
+ },
26
+ {
27
+ "epoch": 1.0,
28
+ "eval_loss": 0.05113217607140541,
29
+ "eval_runtime": 18.3397,
30
+ "eval_samples_per_second": 545.265,
31
+ "eval_steps_per_second": 68.158,
32
+ "step": 1407
33
+ },
34
+ {
35
+ "epoch": 1.0660980810234542,
36
+ "grad_norm": 1.1233466863632202,
37
+ "learning_rate": 1.2897417673537079e-05,
38
+ "loss": 0.06107857894897461,
39
+ "step": 1500
40
+ },
41
+ {
42
+ "epoch": 1.4214641080312722,
43
+ "grad_norm": 0.954052746295929,
44
+ "learning_rate": 1.052831082681829e-05,
45
+ "loss": 0.05122759628295898,
46
+ "step": 2000
47
+ },
48
+ {
49
+ "epoch": 1.7768301350390903,
50
+ "grad_norm": 0.5427104234695435,
51
+ "learning_rate": 8.159203980099502e-06,
52
+ "loss": 0.0449744873046875,
53
+ "step": 2500
54
+ },
55
+ {
56
+ "epoch": 2.0,
57
+ "eval_loss": 0.041359808295965195,
58
+ "eval_runtime": 18.3676,
59
+ "eval_samples_per_second": 544.436,
60
+ "eval_steps_per_second": 68.055,
61
+ "step": 2814
62
+ },
63
+ {
64
+ "epoch": 2.1321961620469083,
65
+ "grad_norm": 0.6617690324783325,
66
+ "learning_rate": 5.790097133380715e-06,
67
+ "loss": 0.041459095001220705,
68
+ "step": 3000
69
+ },
70
+ {
71
+ "epoch": 2.487562189054726,
72
+ "grad_norm": 0.4485235810279846,
73
+ "learning_rate": 3.4209902866619288e-06,
74
+ "loss": 0.036558002471923826,
75
+ "step": 3500
76
+ },
77
+ {
78
+ "epoch": 2.8429282160625444,
79
+ "grad_norm": 0.5422578454017639,
80
+ "learning_rate": 1.0518834399431415e-06,
81
+ "loss": 0.03468650054931641,
82
+ "step": 4000
83
+ },
84
+ {
85
+ "epoch": 3.0,
86
+ "eval_loss": 0.0381973534822464,
87
+ "eval_runtime": 19.0906,
88
+ "eval_samples_per_second": 523.819,
89
+ "eval_steps_per_second": 65.477,
90
+ "step": 4221
91
+ }
92
+ ],
93
+ "logging_steps": 500,
94
+ "max_steps": 4221,
95
+ "num_input_tokens_seen": 0,
96
+ "num_train_epochs": 3,
97
+ "save_steps": 500,
98
+ "stateful_callbacks": {
99
+ "TrainerControl": {
100
+ "args": {
101
+ "should_epoch_stop": false,
102
+ "should_evaluate": false,
103
+ "should_log": false,
104
+ "should_save": true,
105
+ "should_training_stop": true
106
+ },
107
+ "attributes": {}
108
+ }
109
+ },
110
+ "total_flos": 7.06030649856e+16,
111
+ "train_batch_size": 64,
112
+ "trial_name": null,
113
+ "trial_params": null
114
+ }
gretel-pii-ready-augmented/TokenBased-BERT/checkpoint-4221/training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:54c2c0c4bc66849033182b83cc6ed373239dd57860c0cb28f90d25bee0ef7ded
3
+ size 5201
gretel-pii-ready-augmented/TokenBased-BERT/config.json ADDED
@@ -0,0 +1,208 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "add_cross_attention": false,
3
+ "architectures": [
4
+ "BertForTokenClassification"
5
+ ],
6
+ "attention_probs_dropout_prob": 0.1,
7
+ "bos_token_id": null,
8
+ "classifier_dropout": null,
9
+ "directionality": "bidi",
10
+ "dtype": "float32",
11
+ "eos_token_id": null,
12
+ "hidden_act": "gelu",
13
+ "hidden_dropout_prob": 0.1,
14
+ "hidden_size": 768,
15
+ "id2label": {
16
+ "0": "O",
17
+ "1": "B-ACCOUNT_NUMBER",
18
+ "2": "I-ACCOUNT_NUMBER",
19
+ "3": "B-ADDRESS",
20
+ "4": "I-ADDRESS",
21
+ "5": "B-API_KEY",
22
+ "6": "I-API_KEY",
23
+ "7": "B-BANK_ROUTING_NUMBER",
24
+ "8": "I-BANK_ROUTING_NUMBER",
25
+ "9": "B-BIOMETRIC_IDENTIFIER",
26
+ "10": "I-BIOMETRIC_IDENTIFIER",
27
+ "11": "B-CERTIFICATE_LICENSE_NUMBER",
28
+ "12": "I-CERTIFICATE_LICENSE_NUMBER",
29
+ "13": "B-CITY",
30
+ "14": "I-CITY",
31
+ "15": "B-COMPANY_NAME",
32
+ "16": "I-COMPANY_NAME",
33
+ "17": "B-COORDINATE",
34
+ "18": "I-COORDINATE",
35
+ "19": "B-COUNTRY",
36
+ "20": "I-COUNTRY",
37
+ "21": "B-CREDIT_CARD_NUMBER",
38
+ "22": "I-CREDIT_CARD_NUMBER",
39
+ "23": "B-CUSTOMER_ID",
40
+ "24": "I-CUSTOMER_ID",
41
+ "25": "B-CVV",
42
+ "26": "I-CVV",
43
+ "27": "B-DATE",
44
+ "28": "I-DATE",
45
+ "29": "B-DATE_OF_BIRTH",
46
+ "30": "I-DATE_OF_BIRTH",
47
+ "31": "B-DATE_TIME",
48
+ "32": "I-DATE_TIME",
49
+ "33": "B-DEVICE_IDENTIFIER",
50
+ "34": "I-DEVICE_IDENTIFIER",
51
+ "35": "B-EMAIL",
52
+ "36": "I-EMAIL",
53
+ "37": "B-EMPLOYEE_ID",
54
+ "38": "I-EMPLOYEE_ID",
55
+ "39": "B-FIRST_NAME",
56
+ "40": "I-FIRST_NAME",
57
+ "41": "B-HEALTH_PLAN_BENEFICIARY_NUMBER",
58
+ "42": "I-HEALTH_PLAN_BENEFICIARY_NUMBER",
59
+ "43": "B-IPV4",
60
+ "44": "I-IPV4",
61
+ "45": "B-IPV6",
62
+ "46": "I-IPV6",
63
+ "47": "B-LAST_NAME",
64
+ "48": "I-LAST_NAME",
65
+ "49": "B-LICENSE_PLATE",
66
+ "50": "I-LICENSE_PLATE",
67
+ "51": "B-MEDICAL_RECORD_NUMBER",
68
+ "52": "I-MEDICAL_RECORD_NUMBER",
69
+ "53": "B-NAME",
70
+ "54": "I-NAME",
71
+ "55": "B-NATIONAL_ID",
72
+ "56": "I-NATIONAL_ID",
73
+ "57": "B-PASSWORD",
74
+ "58": "I-PASSWORD",
75
+ "59": "B-PHONE_NUMBER",
76
+ "60": "I-PHONE_NUMBER",
77
+ "61": "B-PIN",
78
+ "62": "I-PIN",
79
+ "63": "B-POSTCODE",
80
+ "64": "I-POSTCODE",
81
+ "65": "B-SSN",
82
+ "66": "I-SSN",
83
+ "67": "B-STATE",
84
+ "68": "I-STATE",
85
+ "69": "B-STREET_ADDRESS",
86
+ "70": "I-STREET_ADDRESS",
87
+ "71": "B-SWIFT_BIC",
88
+ "72": "I-SWIFT_BIC",
89
+ "73": "B-TAX_ID",
90
+ "74": "I-TAX_ID",
91
+ "75": "B-TIME",
92
+ "76": "I-TIME",
93
+ "77": "B-UNIQUE_IDENTIFIER",
94
+ "78": "I-UNIQUE_IDENTIFIER",
95
+ "79": "B-URL",
96
+ "80": "I-URL",
97
+ "81": "B-USER_NAME",
98
+ "82": "I-USER_NAME",
99
+ "83": "B-VEHICLE_IDENTIFIER",
100
+ "84": "I-VEHICLE_IDENTIFIER"
101
+ },
102
+ "initializer_range": 0.02,
103
+ "intermediate_size": 3072,
104
+ "is_decoder": false,
105
+ "label2id": {
106
+ "B-ACCOUNT_NUMBER": 1,
107
+ "B-ADDRESS": 3,
108
+ "B-API_KEY": 5,
109
+ "B-BANK_ROUTING_NUMBER": 7,
110
+ "B-BIOMETRIC_IDENTIFIER": 9,
111
+ "B-CERTIFICATE_LICENSE_NUMBER": 11,
112
+ "B-CITY": 13,
113
+ "B-COMPANY_NAME": 15,
114
+ "B-COORDINATE": 17,
115
+ "B-COUNTRY": 19,
116
+ "B-CREDIT_CARD_NUMBER": 21,
117
+ "B-CUSTOMER_ID": 23,
118
+ "B-CVV": 25,
119
+ "B-DATE": 27,
120
+ "B-DATE_OF_BIRTH": 29,
121
+ "B-DATE_TIME": 31,
122
+ "B-DEVICE_IDENTIFIER": 33,
123
+ "B-EMAIL": 35,
124
+ "B-EMPLOYEE_ID": 37,
125
+ "B-FIRST_NAME": 39,
126
+ "B-HEALTH_PLAN_BENEFICIARY_NUMBER": 41,
127
+ "B-IPV4": 43,
128
+ "B-IPV6": 45,
129
+ "B-LAST_NAME": 47,
130
+ "B-LICENSE_PLATE": 49,
131
+ "B-MEDICAL_RECORD_NUMBER": 51,
132
+ "B-NAME": 53,
133
+ "B-NATIONAL_ID": 55,
134
+ "B-PASSWORD": 57,
135
+ "B-PHONE_NUMBER": 59,
136
+ "B-PIN": 61,
137
+ "B-POSTCODE": 63,
138
+ "B-SSN": 65,
139
+ "B-STATE": 67,
140
+ "B-STREET_ADDRESS": 69,
141
+ "B-SWIFT_BIC": 71,
142
+ "B-TAX_ID": 73,
143
+ "B-TIME": 75,
144
+ "B-UNIQUE_IDENTIFIER": 77,
145
+ "B-URL": 79,
146
+ "B-USER_NAME": 81,
147
+ "B-VEHICLE_IDENTIFIER": 83,
148
+ "I-ACCOUNT_NUMBER": 2,
149
+ "I-ADDRESS": 4,
150
+ "I-API_KEY": 6,
151
+ "I-BANK_ROUTING_NUMBER": 8,
152
+ "I-BIOMETRIC_IDENTIFIER": 10,
153
+ "I-CERTIFICATE_LICENSE_NUMBER": 12,
154
+ "I-CITY": 14,
155
+ "I-COMPANY_NAME": 16,
156
+ "I-COORDINATE": 18,
157
+ "I-COUNTRY": 20,
158
+ "I-CREDIT_CARD_NUMBER": 22,
159
+ "I-CUSTOMER_ID": 24,
160
+ "I-CVV": 26,
161
+ "I-DATE": 28,
162
+ "I-DATE_OF_BIRTH": 30,
163
+ "I-DATE_TIME": 32,
164
+ "I-DEVICE_IDENTIFIER": 34,
165
+ "I-EMAIL": 36,
166
+ "I-EMPLOYEE_ID": 38,
167
+ "I-FIRST_NAME": 40,
168
+ "I-HEALTH_PLAN_BENEFICIARY_NUMBER": 42,
169
+ "I-IPV4": 44,
170
+ "I-IPV6": 46,
171
+ "I-LAST_NAME": 48,
172
+ "I-LICENSE_PLATE": 50,
173
+ "I-MEDICAL_RECORD_NUMBER": 52,
174
+ "I-NAME": 54,
175
+ "I-NATIONAL_ID": 56,
176
+ "I-PASSWORD": 58,
177
+ "I-PHONE_NUMBER": 60,
178
+ "I-PIN": 62,
179
+ "I-POSTCODE": 64,
180
+ "I-SSN": 66,
181
+ "I-STATE": 68,
182
+ "I-STREET_ADDRESS": 70,
183
+ "I-SWIFT_BIC": 72,
184
+ "I-TAX_ID": 74,
185
+ "I-TIME": 76,
186
+ "I-UNIQUE_IDENTIFIER": 78,
187
+ "I-URL": 80,
188
+ "I-USER_NAME": 82,
189
+ "I-VEHICLE_IDENTIFIER": 84,
190
+ "O": 0
191
+ },
192
+ "layer_norm_eps": 1e-12,
193
+ "max_position_embeddings": 512,
194
+ "model_type": "bert",
195
+ "num_attention_heads": 12,
196
+ "num_hidden_layers": 12,
197
+ "pad_token_id": 0,
198
+ "pooler_fc_size": 768,
199
+ "pooler_num_attention_heads": 12,
200
+ "pooler_num_fc_layers": 3,
201
+ "pooler_size_per_head": 128,
202
+ "pooler_type": "first_token_transform",
203
+ "tie_word_embeddings": true,
204
+ "transformers_version": "5.3.0",
205
+ "type_vocab_size": 2,
206
+ "use_cache": false,
207
+ "vocab_size": 119547
208
+ }
gretel-pii-ready-augmented/TokenBased-BERT/model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f7c2d3ed068f24221ef2a2864932176a9fc54ebf7fa71b26bfa4ece39f72bcda
3
+ size 709336188
gretel-pii-ready-augmented/TokenBased-BERT/tokenizer.json ADDED
The diff for this file is too large to render. See raw diff
 
gretel-pii-ready-augmented/TokenBased-BERT/tokenizer_config.json ADDED
@@ -0,0 +1,15 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "add_prefix_space": true,
3
+ "backend": "tokenizers",
4
+ "cls_token": "[CLS]",
5
+ "do_lower_case": false,
6
+ "is_local": false,
7
+ "mask_token": "[MASK]",
8
+ "model_max_length": 512,
9
+ "pad_token": "[PAD]",
10
+ "sep_token": "[SEP]",
11
+ "strip_accents": null,
12
+ "tokenize_chinese_chars": true,
13
+ "tokenizer_class": "BertTokenizer",
14
+ "unk_token": "[UNK]"
15
+ }
gretel-pii-ready-augmented/TokenBased-BERT/training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:54c2c0c4bc66849033182b83cc6ed373239dd57860c0cb28f90d25bee0ef7ded
3
+ size 5201