Token Classification
Transformers
Safetensors
PyTorch
Arabic
qwen3
ner
pii
pii-detection
de-identification
privacy
healthcare
medical
clinical
phi
arabic
openmed
Eval Results (legacy)
text-generation-inference
Instructions to use OpenMed/OpenMed-PII-Japanese-QwenMed-XLarge-600M-v1 with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use OpenMed/OpenMed-PII-Japanese-QwenMed-XLarge-600M-v1 with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("token-classification", model="OpenMed/OpenMed-PII-Japanese-QwenMed-XLarge-600M-v1")# Load model directly from transformers import AutoTokenizer, AutoModelForTokenClassification tokenizer = AutoTokenizer.from_pretrained("OpenMed/OpenMed-PII-Japanese-QwenMed-XLarge-600M-v1") model = AutoModelForTokenClassification.from_pretrained("OpenMed/OpenMed-PII-Japanese-QwenMed-XLarge-600M-v1", device_map="auto") - Notebooks
- Google Colab
- Kaggle
Upload Japanese PII detection model OpenMed-PII-Japanese-QwenMed-XLarge-600M-v1
Browse files- .gitattributes +1 -0
- README.md +302 -0
- all_results.json +28 -0
- chat_template.jinja +85 -0
- classification_report.txt +55 -0
- config.json +219 -0
- eval_results.json +13 -0
- model.safetensors +3 -0
- test_results.json +12 -0
- tokenizer.json +3 -0
- tokenizer_config.json +14 -0
- train_results.json +8 -0
.gitattributes
CHANGED
|
@@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
|
|
| 33 |
*.zip filter=lfs diff=lfs merge=lfs -text
|
| 34 |
*.zst filter=lfs diff=lfs merge=lfs -text
|
| 35 |
*tfevents* filter=lfs diff=lfs merge=lfs -text
|
|
|
|
|
|
| 33 |
*.zip filter=lfs diff=lfs merge=lfs -text
|
| 34 |
*.zst filter=lfs diff=lfs merge=lfs -text
|
| 35 |
*tfevents* filter=lfs diff=lfs merge=lfs -text
|
| 36 |
+
tokenizer.json filter=lfs diff=lfs merge=lfs -text
|
README.md
ADDED
|
@@ -0,0 +1,302 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
---
|
| 2 |
+
language:
|
| 3 |
+
- ar
|
| 4 |
+
license: apache-2.0
|
| 5 |
+
base_model: Qwen/Qwen3-Embedding-0.6B
|
| 6 |
+
tags:
|
| 7 |
+
- token-classification
|
| 8 |
+
- ner
|
| 9 |
+
- pii
|
| 10 |
+
- pii-detection
|
| 11 |
+
- de-identification
|
| 12 |
+
- privacy
|
| 13 |
+
- healthcare
|
| 14 |
+
- medical
|
| 15 |
+
- clinical
|
| 16 |
+
- phi
|
| 17 |
+
- arabic
|
| 18 |
+
- pytorch
|
| 19 |
+
- transformers
|
| 20 |
+
- openmed
|
| 21 |
+
pipeline_tag: token-classification
|
| 22 |
+
library_name: transformers
|
| 23 |
+
metrics:
|
| 24 |
+
- f1
|
| 25 |
+
- precision
|
| 26 |
+
- recall
|
| 27 |
+
model-index:
|
| 28 |
+
- name: OpenMed-PII-Arabic-QwenMed-XLarge-600M-v1
|
| 29 |
+
results:
|
| 30 |
+
- task:
|
| 31 |
+
type: token-classification
|
| 32 |
+
name: Named Entity Recognition
|
| 33 |
+
dataset:
|
| 34 |
+
name: AI4Privacy + Synthetic Arabic PII
|
| 35 |
+
type: ai4privacy/pii-masking-200k
|
| 36 |
+
split: test
|
| 37 |
+
metrics:
|
| 38 |
+
- type: f1
|
| 39 |
+
value: 0.7929
|
| 40 |
+
name: F1 (micro)
|
| 41 |
+
- type: precision
|
| 42 |
+
value: 0.7832
|
| 43 |
+
name: Precision
|
| 44 |
+
- type: recall
|
| 45 |
+
value: 0.8028
|
| 46 |
+
name: Recall
|
| 47 |
+
widget:
|
| 48 |
+
- text: "د. أحمد محمد (رقم الهوية: 1234567890) يمكن التواصل معه عبر ahmed.mohammed@hospital.sa أو +966 50 123 4567. العنوان: شارع الملك فهد 25، الرياض 11564."
|
| 49 |
+
example_title: Clinical Note with PII (Arabic)
|
| 50 |
+
---
|
| 51 |
+
|
| 52 |
+
# OpenMed-PII-Arabic-QwenMed-XLarge-600M-v1
|
| 53 |
+
|
| 54 |
+
**Arabic PII Detection Model** | 600M Parameters | Open Source
|
| 55 |
+
|
| 56 |
+
[]() []() []()
|
| 57 |
+
|
| 58 |
+
## Model Description
|
| 59 |
+
|
| 60 |
+
**OpenMed-PII-Arabic-QwenMed-XLarge-600M-v1** is a transformer-based token classification model fine-tuned for **Personally Identifiable Information (PII) detection in Arabic text**. This model identifies and classifies **54 types of sensitive information** including names, addresses, social security numbers, medical record numbers, and more.
|
| 61 |
+
|
| 62 |
+
### Key Features
|
| 63 |
+
|
| 64 |
+
- **Arabic-Optimized**: Specifically trained on Arabic text for optimal performance
|
| 65 |
+
- **High Accuracy**: Achieves strong F1 scores across diverse PII categories
|
| 66 |
+
- **Comprehensive Coverage**: Detects 55+ entity types spanning personal, financial, medical, and contact information
|
| 67 |
+
- **Privacy-Focused**: Designed for de-identification and compliance with GDPR and other privacy regulations
|
| 68 |
+
- **Production-Ready**: Optimized for real-world text processing pipelines
|
| 69 |
+
|
| 70 |
+
## Performance
|
| 71 |
+
|
| 72 |
+
Evaluated on the Arabic test split (AI4Privacy + synthetic data):
|
| 73 |
+
|
| 74 |
+
| Metric | Score |
|
| 75 |
+
|:---|:---:|
|
| 76 |
+
| **Micro F1** | **0.7929** |
|
| 77 |
+
| Precision | 0.7832 |
|
| 78 |
+
| Recall | 0.8028 |
|
| 79 |
+
| Macro F1 | 0.4080 |
|
| 80 |
+
| Weighted F1 | 0.8030 |
|
| 81 |
+
| Accuracy | 0.8920 |
|
| 82 |
+
|
| 83 |
+
### Top 3 Arabic PII Models
|
| 84 |
+
|
| 85 |
+
| Rank | Model | F1 | Precision | Recall |
|
| 86 |
+
|:---:|:---|:---:|:---:|:---:|
|
| 87 |
+
| 1 | [OpenMed-PII-Arabic-BigMed-Large-560M-v1](https://huggingface.co/OpenMed/OpenMed-PII-Arabic-BigMed-Large-560M-v1) | 0.8410 | 0.8531 | 0.8292 |
|
| 88 |
+
| 2 | [OpenMed-PII-Arabic-NomicMed-Large-395M-v1](https://huggingface.co/OpenMed/OpenMed-PII-Arabic-NomicMed-Large-395M-v1) | 0.8281 | 0.8126 | 0.8441 |
|
| 89 |
+
| **3** | **[OpenMed-PII-Arabic-QwenMed-XLarge-600M-v1](https://huggingface.co/OpenMed/OpenMed-PII-Arabic-QwenMed-XLarge-600M-v1)** | **0.7929** | **0.7832** | **0.8028** |
|
| 90 |
+
|
| 91 |
+
## Supported Entity Types
|
| 92 |
+
|
| 93 |
+
This model detects **54 PII entity types** organized into categories:
|
| 94 |
+
|
| 95 |
+
<details>
|
| 96 |
+
<summary><strong>Identifiers</strong> (22 types)</summary>
|
| 97 |
+
|
| 98 |
+
| Entity | Description |
|
| 99 |
+
|:---|:---|
|
| 100 |
+
| `ACCOUNTNAME` | Accountname |
|
| 101 |
+
| `BANKACCOUNT` | Bankaccount |
|
| 102 |
+
| `BIC` | Bic |
|
| 103 |
+
| `BITCOINADDRESS` | Bitcoinaddress |
|
| 104 |
+
| `CREDITCARD` | Creditcard |
|
| 105 |
+
| `CREDITCARDISSUER` | Creditcardissuer |
|
| 106 |
+
| `CVV` | Cvv |
|
| 107 |
+
| `ETHEREUMADDRESS` | Ethereumaddress |
|
| 108 |
+
| `IBAN` | Iban |
|
| 109 |
+
| `IMEI` | Imei |
|
| 110 |
+
| ... | *and 12 more* |
|
| 111 |
+
|
| 112 |
+
</details>
|
| 113 |
+
|
| 114 |
+
<details>
|
| 115 |
+
<summary><strong>Personal Info</strong> (11 types)</summary>
|
| 116 |
+
|
| 117 |
+
| Entity | Description |
|
| 118 |
+
|:---|:---|
|
| 119 |
+
| `AGE` | Age |
|
| 120 |
+
| `DATEOFBIRTH` | Dateofbirth |
|
| 121 |
+
| `EYECOLOR` | Eyecolor |
|
| 122 |
+
| `FIRSTNAME` | Firstname |
|
| 123 |
+
| `GENDER` | Gender |
|
| 124 |
+
| `HEIGHT` | Height |
|
| 125 |
+
| `LASTNAME` | Lastname |
|
| 126 |
+
| `MIDDLENAME` | Middlename |
|
| 127 |
+
| `OCCUPATION` | Occupation |
|
| 128 |
+
| `PREFIX` | Prefix |
|
| 129 |
+
| ... | *and 1 more* |
|
| 130 |
+
|
| 131 |
+
</details>
|
| 132 |
+
|
| 133 |
+
<details>
|
| 134 |
+
<summary><strong>Contact Info</strong> (2 types)</summary>
|
| 135 |
+
|
| 136 |
+
| Entity | Description |
|
| 137 |
+
|:---|:---|
|
| 138 |
+
| `EMAIL` | Email |
|
| 139 |
+
| `PHONE` | Phone |
|
| 140 |
+
|
| 141 |
+
</details>
|
| 142 |
+
|
| 143 |
+
<details>
|
| 144 |
+
<summary><strong>Location</strong> (9 types)</summary>
|
| 145 |
+
|
| 146 |
+
| Entity | Description |
|
| 147 |
+
|:---|:---|
|
| 148 |
+
| `BUILDINGNUMBER` | Buildingnumber |
|
| 149 |
+
| `CITY` | City |
|
| 150 |
+
| `COUNTY` | County |
|
| 151 |
+
| `GPSCOORDINATES` | Gpscoordinates |
|
| 152 |
+
| `ORDINALDIRECTION` | Ordinaldirection |
|
| 153 |
+
| `SECONDARYADDRESS` | Secondaryaddress |
|
| 154 |
+
| `STATE` | State |
|
| 155 |
+
| `STREET` | Street |
|
| 156 |
+
| `ZIPCODE` | Zipcode |
|
| 157 |
+
|
| 158 |
+
</details>
|
| 159 |
+
|
| 160 |
+
<details>
|
| 161 |
+
<summary><strong>Organization</strong> (3 types)</summary>
|
| 162 |
+
|
| 163 |
+
| Entity | Description |
|
| 164 |
+
|:---|:---|
|
| 165 |
+
| `JOBDEPARTMENT` | Jobdepartment |
|
| 166 |
+
| `JOBTITLE` | Jobtitle |
|
| 167 |
+
| `ORGANIZATION` | Organization |
|
| 168 |
+
|
| 169 |
+
</details>
|
| 170 |
+
|
| 171 |
+
<details>
|
| 172 |
+
<summary><strong>Financial</strong> (5 types)</summary>
|
| 173 |
+
|
| 174 |
+
| Entity | Description |
|
| 175 |
+
|:---|:---|
|
| 176 |
+
| `AMOUNT` | Amount |
|
| 177 |
+
| `CURRENCY` | Currency |
|
| 178 |
+
| `CURRENCYCODE` | Currencycode |
|
| 179 |
+
| `CURRENCYNAME` | Currencyname |
|
| 180 |
+
| `CURRENCYSYMBOL` | Currencysymbol |
|
| 181 |
+
|
| 182 |
+
</details>
|
| 183 |
+
|
| 184 |
+
<details>
|
| 185 |
+
<summary><strong>Temporal</strong> (2 types)</summary>
|
| 186 |
+
|
| 187 |
+
| Entity | Description |
|
| 188 |
+
|:---|:---|
|
| 189 |
+
| `DATE` | Date |
|
| 190 |
+
| `TIME` | Time |
|
| 191 |
+
|
| 192 |
+
</details>
|
| 193 |
+
|
| 194 |
+
## Usage
|
| 195 |
+
|
| 196 |
+
### Quick Start
|
| 197 |
+
|
| 198 |
+
```python
|
| 199 |
+
from transformers import pipeline
|
| 200 |
+
|
| 201 |
+
# Load the PII detection pipeline
|
| 202 |
+
ner = pipeline("ner", model="OpenMed/OpenMed-PII-Arabic-QwenMed-XLarge-600M-v1", aggregation_strategy="simple")
|
| 203 |
+
|
| 204 |
+
text = """
|
| 205 |
+
المريض خالد العتيبي (تاريخ الميلاد: 15/03/1985، رقم الهوية: 9876543210) تم فحصه اليوم.
|
| 206 |
+
التواصل: khaled.otaibi@email.sa، الهاتف: +966 50 123 4567.
|
| 207 |
+
العنوان: شارع العليا 42، الرياض 11432.
|
| 208 |
+
"""
|
| 209 |
+
|
| 210 |
+
entities = ner(text)
|
| 211 |
+
for entity in entities:
|
| 212 |
+
print(f"{entity['entity_group']}: {entity['word']} (score: {entity['score']:.3f})")
|
| 213 |
+
```
|
| 214 |
+
|
| 215 |
+
### De-identification Example
|
| 216 |
+
|
| 217 |
+
```python
|
| 218 |
+
def redact_pii(text, entities, placeholder='[REDACTED]'):
|
| 219 |
+
"""Replace detected PII with placeholders."""
|
| 220 |
+
# Sort entities by start position (descending) to preserve offsets
|
| 221 |
+
sorted_entities = sorted(entities, key=lambda x: x['start'], reverse=True)
|
| 222 |
+
redacted = text
|
| 223 |
+
for ent in sorted_entities:
|
| 224 |
+
redacted = redacted[:ent['start']] + f"[{ent['entity_group']}]" + redacted[ent['end']:]
|
| 225 |
+
return redacted
|
| 226 |
+
|
| 227 |
+
# Apply de-identification
|
| 228 |
+
redacted_text = redact_pii(text, entities)
|
| 229 |
+
print(redacted_text)
|
| 230 |
+
```
|
| 231 |
+
|
| 232 |
+
### Batch Processing
|
| 233 |
+
|
| 234 |
+
```python
|
| 235 |
+
from transformers import AutoModelForTokenClassification, AutoTokenizer
|
| 236 |
+
import torch
|
| 237 |
+
|
| 238 |
+
model_name = "OpenMed/OpenMed-PII-Arabic-QwenMed-XLarge-600M-v1"
|
| 239 |
+
model = AutoModelForTokenClassification.from_pretrained(model_name)
|
| 240 |
+
tokenizer = AutoTokenizer.from_pretrained(model_name)
|
| 241 |
+
|
| 242 |
+
texts = [
|
| 243 |
+
"المريض خالد العتيبي (تاريخ الميلاد: 15/03/1985، رقم الهوية: 9876543210) تم فحصه اليوم.",
|
| 244 |
+
"التواصل: khaled.otaibi@email.sa، الهاتف: +966 50 123 4567.",
|
| 245 |
+
]
|
| 246 |
+
|
| 247 |
+
inputs = tokenizer(texts, return_tensors='pt', padding=True, truncation=True)
|
| 248 |
+
with torch.no_grad():
|
| 249 |
+
outputs = model(**inputs)
|
| 250 |
+
predictions = torch.argmax(outputs.logits, dim=-1)
|
| 251 |
+
```
|
| 252 |
+
|
| 253 |
+
## Training Details
|
| 254 |
+
|
| 255 |
+
### Dataset
|
| 256 |
+
|
| 257 |
+
This model was trained on a combination of:
|
| 258 |
+
|
| 259 |
+
- **[AI4Privacy PII Masking 200K](https://huggingface.co/datasets/ai4privacy/pii-masking-200k)**: Multilingual base dataset (200K records across 8 languages)
|
| 260 |
+
- **[NVIDIA Nemotron-PII](https://huggingface.co/datasets/nvidia/Nemotron-PII)**: Seed dataset for synthetic data generation
|
| 261 |
+
- **Synthetic Arabic Data**: ~25K high-quality samples generated with locale-specific formatting (National ID format, +966 phones, Arabic names, SAR/ر.س currency)
|
| 262 |
+
|
| 263 |
+
- **Format**: BIO-tagged token classification
|
| 264 |
+
- **Labels**: 76 BIO tags (54 entity types)
|
| 265 |
+
|
| 266 |
+
### Training Configuration
|
| 267 |
+
|
| 268 |
+
- **Max Sequence Length**: 512 tokens
|
| 269 |
+
- **Framework**: Hugging Face Transformers + Trainer API
|
| 270 |
+
|
| 271 |
+
## Intended Use & Limitations
|
| 272 |
+
|
| 273 |
+
### Intended Use
|
| 274 |
+
|
| 275 |
+
- **De-identification**: Automated redaction of PII in Arabic clinical notes, medical records, and documents
|
| 276 |
+
- **Compliance**: Supporting GDPR, and other privacy regulation compliance
|
| 277 |
+
- **Data Preprocessing**: Preparing datasets for research by removing sensitive information
|
| 278 |
+
- **Audit Support**: Identifying PII in document collections
|
| 279 |
+
|
| 280 |
+
### Limitations
|
| 281 |
+
|
| 282 |
+
**Important**: This model is intended as an **assistive tool**, not a replacement for human review.
|
| 283 |
+
|
| 284 |
+
- **False Negatives**: Some PII may not be detected; always verify critical applications
|
| 285 |
+
- **Context Sensitivity**: Performance may vary with domain-specific terminology
|
| 286 |
+
- **Language**: Optimized for Arabic text; may not perform well on other languages
|
| 287 |
+
|
| 288 |
+
## Citation
|
| 289 |
+
|
| 290 |
+
```bibtex
|
| 291 |
+
@misc{openmed-pii-2026,
|
| 292 |
+
title = {OpenMed-PII-Arabic-QwenMed-XLarge-600M-v1: Arabic PII Detection Model},
|
| 293 |
+
author = {OpenMed Science},
|
| 294 |
+
year = {2026},
|
| 295 |
+
publisher = {Hugging Face},
|
| 296 |
+
url = {https://huggingface.co/OpenMed/OpenMed-PII-Arabic-QwenMed-XLarge-600M-v1}
|
| 297 |
+
}
|
| 298 |
+
```
|
| 299 |
+
|
| 300 |
+
## Links
|
| 301 |
+
|
| 302 |
+
- **Organization**: [OpenMed](https://huggingface.co/OpenMed)
|
all_results.json
ADDED
|
@@ -0,0 +1,28 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"epoch": 3.0,
|
| 3 |
+
"eval_accuracy": 0.888999548715395,
|
| 4 |
+
"eval_f1": 0.7939432989690722,
|
| 5 |
+
"eval_loss": 0.5661426782608032,
|
| 6 |
+
"eval_macro_f1": 0.3903729497661312,
|
| 7 |
+
"eval_precision": 0.7801202912314024,
|
| 8 |
+
"eval_recall": 0.8082650049196458,
|
| 9 |
+
"eval_runtime": 2.8617,
|
| 10 |
+
"eval_samples_per_second": 535.703,
|
| 11 |
+
"eval_steps_per_second": 16.773,
|
| 12 |
+
"eval_weighted_f1": 0.8033758003197404,
|
| 13 |
+
"test_accuracy": 0.8919662851455586,
|
| 14 |
+
"test_f1": 0.7929066580686763,
|
| 15 |
+
"test_loss": 0.5452415943145752,
|
| 16 |
+
"test_macro_f1": 0.40795926781161085,
|
| 17 |
+
"test_precision": 0.7832346009299955,
|
| 18 |
+
"test_recall": 0.8028205797858449,
|
| 19 |
+
"test_runtime": 2.8812,
|
| 20 |
+
"test_samples_per_second": 532.077,
|
| 21 |
+
"test_steps_per_second": 16.66,
|
| 22 |
+
"test_weighted_f1": 0.803044059461668,
|
| 23 |
+
"total_flos": 5481934918516736.0,
|
| 24 |
+
"train_loss": 3.2885495954089694,
|
| 25 |
+
"train_runtime": 335.4941,
|
| 26 |
+
"train_samples_per_second": 109.683,
|
| 27 |
+
"train_steps_per_second": 1.717
|
| 28 |
+
}
|
chat_template.jinja
ADDED
|
@@ -0,0 +1,85 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{%- if tools %}
|
| 2 |
+
{{- '<|im_start|>system\n' }}
|
| 3 |
+
{%- if messages[0].role == 'system' %}
|
| 4 |
+
{{- messages[0].content + '\n\n' }}
|
| 5 |
+
{%- endif %}
|
| 6 |
+
{{- "# Tools\n\nYou may call one or more functions to assist with the user query.\n\nYou are provided with function signatures within <tools></tools> XML tags:\n<tools>" }}
|
| 7 |
+
{%- for tool in tools %}
|
| 8 |
+
{{- "\n" }}
|
| 9 |
+
{{- tool | tojson }}
|
| 10 |
+
{%- endfor %}
|
| 11 |
+
{{- "\n</tools>\n\nFor each function call, return a json object with function name and arguments within <tool_call></tool_call> XML tags:\n<tool_call>\n{\"name\": <function-name>, \"arguments\": <args-json-object>}\n</tool_call><|im_end|>\n" }}
|
| 12 |
+
{%- else %}
|
| 13 |
+
{%- if messages[0].role == 'system' %}
|
| 14 |
+
{{- '<|im_start|>system\n' + messages[0].content + '<|im_end|>\n' }}
|
| 15 |
+
{%- endif %}
|
| 16 |
+
{%- endif %}
|
| 17 |
+
{%- set ns = namespace(multi_step_tool=true, last_query_index=messages|length - 1) %}
|
| 18 |
+
{%- for message in messages[::-1] %}
|
| 19 |
+
{%- set index = (messages|length - 1) - loop.index0 %}
|
| 20 |
+
{%- if ns.multi_step_tool and message.role == "user" and not(message.content.startswith('<tool_response>') and message.content.endswith('</tool_response>')) %}
|
| 21 |
+
{%- set ns.multi_step_tool = false %}
|
| 22 |
+
{%- set ns.last_query_index = index %}
|
| 23 |
+
{%- endif %}
|
| 24 |
+
{%- endfor %}
|
| 25 |
+
{%- for message in messages %}
|
| 26 |
+
{%- if (message.role == "user") or (message.role == "system" and not loop.first) %}
|
| 27 |
+
{{- '<|im_start|>' + message.role + '\n' + message.content + '<|im_end|>' + '\n' }}
|
| 28 |
+
{%- elif message.role == "assistant" %}
|
| 29 |
+
{%- set content = message.content %}
|
| 30 |
+
{%- set reasoning_content = '' %}
|
| 31 |
+
{%- if message.reasoning_content is defined and message.reasoning_content is not none %}
|
| 32 |
+
{%- set reasoning_content = message.reasoning_content %}
|
| 33 |
+
{%- else %}
|
| 34 |
+
{%- if '</think>' in message.content %}
|
| 35 |
+
{%- set content = message.content.split('</think>')[-1].lstrip('\n') %}
|
| 36 |
+
{%- set reasoning_content = message.content.split('</think>')[0].rstrip('\n').split('<think>')[-1].lstrip('\n') %}
|
| 37 |
+
{%- endif %}
|
| 38 |
+
{%- endif %}
|
| 39 |
+
{%- if loop.index0 > ns.last_query_index %}
|
| 40 |
+
{%- if loop.last or (not loop.last and reasoning_content) %}
|
| 41 |
+
{{- '<|im_start|>' + message.role + '\n<think>\n' + reasoning_content.strip('\n') + '\n</think>\n\n' + content.lstrip('\n') }}
|
| 42 |
+
{%- else %}
|
| 43 |
+
{{- '<|im_start|>' + message.role + '\n' + content }}
|
| 44 |
+
{%- endif %}
|
| 45 |
+
{%- else %}
|
| 46 |
+
{{- '<|im_start|>' + message.role + '\n' + content }}
|
| 47 |
+
{%- endif %}
|
| 48 |
+
{%- if message.tool_calls %}
|
| 49 |
+
{%- for tool_call in message.tool_calls %}
|
| 50 |
+
{%- if (loop.first and content) or (not loop.first) %}
|
| 51 |
+
{{- '\n' }}
|
| 52 |
+
{%- endif %}
|
| 53 |
+
{%- if tool_call.function %}
|
| 54 |
+
{%- set tool_call = tool_call.function %}
|
| 55 |
+
{%- endif %}
|
| 56 |
+
{{- '<tool_call>\n{"name": "' }}
|
| 57 |
+
{{- tool_call.name }}
|
| 58 |
+
{{- '", "arguments": ' }}
|
| 59 |
+
{%- if tool_call.arguments is string %}
|
| 60 |
+
{{- tool_call.arguments }}
|
| 61 |
+
{%- else %}
|
| 62 |
+
{{- tool_call.arguments | tojson }}
|
| 63 |
+
{%- endif %}
|
| 64 |
+
{{- '}\n</tool_call>' }}
|
| 65 |
+
{%- endfor %}
|
| 66 |
+
{%- endif %}
|
| 67 |
+
{{- '<|im_end|>\n' }}
|
| 68 |
+
{%- elif message.role == "tool" %}
|
| 69 |
+
{%- if loop.first or (messages[loop.index0 - 1].role != "tool") %}
|
| 70 |
+
{{- '<|im_start|>user' }}
|
| 71 |
+
{%- endif %}
|
| 72 |
+
{{- '\n<tool_response>\n' }}
|
| 73 |
+
{{- message.content }}
|
| 74 |
+
{{- '\n</tool_response>' }}
|
| 75 |
+
{%- if loop.last or (messages[loop.index0 + 1].role != "tool") %}
|
| 76 |
+
{{- '<|im_end|>\n' }}
|
| 77 |
+
{%- endif %}
|
| 78 |
+
{%- endif %}
|
| 79 |
+
{%- endfor %}
|
| 80 |
+
{%- if add_generation_prompt %}
|
| 81 |
+
{{- '<|im_start|>assistant\n' }}
|
| 82 |
+
{%- if enable_thinking is defined and enable_thinking is false %}
|
| 83 |
+
{{- '<think>\n\n</think>\n\n' }}
|
| 84 |
+
{%- endif %}
|
| 85 |
+
{%- endif %}
|
classification_report.txt
ADDED
|
@@ -0,0 +1,55 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
Classification Report for Japanese PII Detection
|
| 2 |
+
Model: Qwen/Qwen3-Embedding-0.6B
|
| 3 |
+
============================================================
|
| 4 |
+
|
| 5 |
+
precision recall f1-score support
|
| 6 |
+
|
| 7 |
+
ACCOUNTNAME 0.28 0.49 0.35 117
|
| 8 |
+
AGE 0.65 0.65 0.65 202
|
| 9 |
+
AMOUNT 0.22 0.27 0.24 385
|
| 10 |
+
BANKACCOUNT 0.85 0.83 0.84 594
|
| 11 |
+
BIC 0.00 0.00 0.00 0
|
| 12 |
+
BITCOINADDRESS 0.00 0.00 0.00 34
|
| 13 |
+
BUILDINGNUMBER 0.35 0.52 0.41 283
|
| 14 |
+
CITY 0.25 0.39 0.30 275
|
| 15 |
+
COUNTY 0.24 0.61 0.34 90
|
| 16 |
+
CREDITCARD 1.00 0.58 0.73 26
|
| 17 |
+
CURRENCY 0.00 0.00 0.00 86
|
| 18 |
+
CURRENCYCODE 0.12 0.50 0.20 12
|
| 19 |
+
CURRENCYSYMBOL 0.60 0.07 0.12 46
|
| 20 |
+
DATE 0.49 0.52 0.51 546
|
| 21 |
+
DATEOFBIRTH 0.61 0.61 0.61 101
|
| 22 |
+
EMAIL 0.98 0.98 0.98 1913
|
| 23 |
+
EYECOLOR 0.00 0.00 0.00 1
|
| 24 |
+
FIRSTNAME 0.99 0.99 0.99 3107
|
| 25 |
+
GENDER 0.36 0.13 0.19 39
|
| 26 |
+
HEIGHT 0.27 0.26 0.27 54
|
| 27 |
+
IBAN 0.72 0.79 0.76 239
|
| 28 |
+
IMEI 0.00 0.00 0.00 15
|
| 29 |
+
IPADDRESS 1.00 0.87 0.93 15
|
| 30 |
+
JOBDEPARTMENT 0.69 0.88 0.77 136
|
| 31 |
+
JOBTITLE 0.27 0.31 0.29 72
|
| 32 |
+
LASTNAME 0.99 0.93 0.96 3227
|
| 33 |
+
MASKEDNUMBER 0.72 0.12 0.20 179
|
| 34 |
+
MIDDLENAME 0.00 0.00 0.00 2
|
| 35 |
+
OCCUPATION 0.31 0.04 0.08 90
|
| 36 |
+
ORDINALDIRECTION 0.00 0.00 0.00 41
|
| 37 |
+
ORGANIZATION 0.32 0.35 0.34 265
|
| 38 |
+
PASSWORD 0.70 0.88 0.78 26
|
| 39 |
+
PHONE 0.95 0.97 0.96 306
|
| 40 |
+
PREFIX 0.93 0.84 0.88 1573
|
| 41 |
+
SECONDARYADDRESS 0.00 0.00 0.00 20
|
| 42 |
+
SEX 0.00 0.00 0.00 9
|
| 43 |
+
SSN 0.86 0.91 0.88 317
|
| 44 |
+
STATE 0.00 0.00 0.00 13
|
| 45 |
+
STREET 0.12 0.23 0.16 192
|
| 46 |
+
TIME 0.36 0.54 0.44 228
|
| 47 |
+
URL 0.92 0.76 0.83 45
|
| 48 |
+
USERNAME 0.50 0.44 0.47 133
|
| 49 |
+
VIN 0.00 0.00 0.00 9
|
| 50 |
+
VRM 0.00 0.00 0.00 16
|
| 51 |
+
ZIPCODE 0.90 0.91 0.91 237
|
| 52 |
+
|
| 53 |
+
micro avg 0.78 0.80 0.79 15316
|
| 54 |
+
macro avg 0.43 0.43 0.41 15316
|
| 55 |
+
weighted avg 0.82 0.80 0.80 15316
|
config.json
ADDED
|
@@ -0,0 +1,219 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"architectures": [
|
| 3 |
+
"Qwen3ForTokenClassification"
|
| 4 |
+
],
|
| 5 |
+
"attention_bias": false,
|
| 6 |
+
"attention_dropout": 0.0,
|
| 7 |
+
"bos_token_id": null,
|
| 8 |
+
"dtype": "bfloat16",
|
| 9 |
+
"eos_token_id": 151645,
|
| 10 |
+
"head_dim": 128,
|
| 11 |
+
"hidden_act": "silu",
|
| 12 |
+
"hidden_size": 1024,
|
| 13 |
+
"id2label": {
|
| 14 |
+
"0": "O",
|
| 15 |
+
"1": "B-ACCOUNTNAME",
|
| 16 |
+
"2": "B-AGE",
|
| 17 |
+
"3": "B-AMOUNT",
|
| 18 |
+
"4": "B-BANKACCOUNT",
|
| 19 |
+
"5": "B-BIC",
|
| 20 |
+
"6": "B-BITCOINADDRESS",
|
| 21 |
+
"7": "B-BUILDINGNUMBER",
|
| 22 |
+
"8": "B-CITY",
|
| 23 |
+
"9": "B-COUNTY",
|
| 24 |
+
"10": "B-CREDITCARD",
|
| 25 |
+
"11": "B-CREDITCARDISSUER",
|
| 26 |
+
"12": "B-CURRENCY",
|
| 27 |
+
"13": "B-CURRENCYCODE",
|
| 28 |
+
"14": "B-CURRENCYNAME",
|
| 29 |
+
"15": "B-CURRENCYSYMBOL",
|
| 30 |
+
"16": "B-CVV",
|
| 31 |
+
"17": "B-DATE",
|
| 32 |
+
"18": "B-DATEOFBIRTH",
|
| 33 |
+
"19": "B-EMAIL",
|
| 34 |
+
"20": "B-ETHEREUMADDRESS",
|
| 35 |
+
"21": "B-EYECOLOR",
|
| 36 |
+
"22": "B-FIRSTNAME",
|
| 37 |
+
"23": "B-GENDER",
|
| 38 |
+
"24": "B-GPSCOORDINATES",
|
| 39 |
+
"25": "B-HEIGHT",
|
| 40 |
+
"26": "B-IBAN",
|
| 41 |
+
"27": "B-IMEI",
|
| 42 |
+
"28": "B-IPADDRESS",
|
| 43 |
+
"29": "B-JOBDEPARTMENT",
|
| 44 |
+
"30": "B-JOBTITLE",
|
| 45 |
+
"31": "B-LASTNAME",
|
| 46 |
+
"32": "B-LITECOINADDRESS",
|
| 47 |
+
"33": "B-MACADDRESS",
|
| 48 |
+
"34": "B-MASKEDNUMBER",
|
| 49 |
+
"35": "B-MIDDLENAME",
|
| 50 |
+
"36": "B-OCCUPATION",
|
| 51 |
+
"37": "B-ORDINALDIRECTION",
|
| 52 |
+
"38": "B-ORGANIZATION",
|
| 53 |
+
"39": "B-PASSWORD",
|
| 54 |
+
"40": "B-PHONE",
|
| 55 |
+
"41": "B-PIN",
|
| 56 |
+
"42": "B-PREFIX",
|
| 57 |
+
"43": "B-SECONDARYADDRESS",
|
| 58 |
+
"44": "B-SEX",
|
| 59 |
+
"45": "B-SSN",
|
| 60 |
+
"46": "B-STATE",
|
| 61 |
+
"47": "B-STREET",
|
| 62 |
+
"48": "B-TIME",
|
| 63 |
+
"49": "B-URL",
|
| 64 |
+
"50": "B-USERAGENT",
|
| 65 |
+
"51": "B-USERNAME",
|
| 66 |
+
"52": "B-VIN",
|
| 67 |
+
"53": "B-VRM",
|
| 68 |
+
"54": "B-ZIPCODE",
|
| 69 |
+
"55": "I-ACCOUNTNAME",
|
| 70 |
+
"56": "I-AGE",
|
| 71 |
+
"57": "I-AMOUNT",
|
| 72 |
+
"58": "I-CITY",
|
| 73 |
+
"59": "I-COUNTY",
|
| 74 |
+
"60": "I-CURRENCY",
|
| 75 |
+
"61": "I-CURRENCYNAME",
|
| 76 |
+
"62": "I-DATE",
|
| 77 |
+
"63": "I-DATEOFBIRTH",
|
| 78 |
+
"64": "I-EYECOLOR",
|
| 79 |
+
"65": "I-GENDER",
|
| 80 |
+
"66": "I-HEIGHT",
|
| 81 |
+
"67": "I-JOBTITLE",
|
| 82 |
+
"68": "I-ORGANIZATION",
|
| 83 |
+
"69": "I-PHONE",
|
| 84 |
+
"70": "I-SECONDARYADDRESS",
|
| 85 |
+
"71": "I-SSN",
|
| 86 |
+
"72": "I-STATE",
|
| 87 |
+
"73": "I-STREET",
|
| 88 |
+
"74": "I-TIME",
|
| 89 |
+
"75": "I-USERAGENT"
|
| 90 |
+
},
|
| 91 |
+
"initializer_range": 0.02,
|
| 92 |
+
"intermediate_size": 3072,
|
| 93 |
+
"label2id": {
|
| 94 |
+
"B-ACCOUNTNAME": 1,
|
| 95 |
+
"B-AGE": 2,
|
| 96 |
+
"B-AMOUNT": 3,
|
| 97 |
+
"B-BANKACCOUNT": 4,
|
| 98 |
+
"B-BIC": 5,
|
| 99 |
+
"B-BITCOINADDRESS": 6,
|
| 100 |
+
"B-BUILDINGNUMBER": 7,
|
| 101 |
+
"B-CITY": 8,
|
| 102 |
+
"B-COUNTY": 9,
|
| 103 |
+
"B-CREDITCARD": 10,
|
| 104 |
+
"B-CREDITCARDISSUER": 11,
|
| 105 |
+
"B-CURRENCY": 12,
|
| 106 |
+
"B-CURRENCYCODE": 13,
|
| 107 |
+
"B-CURRENCYNAME": 14,
|
| 108 |
+
"B-CURRENCYSYMBOL": 15,
|
| 109 |
+
"B-CVV": 16,
|
| 110 |
+
"B-DATE": 17,
|
| 111 |
+
"B-DATEOFBIRTH": 18,
|
| 112 |
+
"B-EMAIL": 19,
|
| 113 |
+
"B-ETHEREUMADDRESS": 20,
|
| 114 |
+
"B-EYECOLOR": 21,
|
| 115 |
+
"B-FIRSTNAME": 22,
|
| 116 |
+
"B-GENDER": 23,
|
| 117 |
+
"B-GPSCOORDINATES": 24,
|
| 118 |
+
"B-HEIGHT": 25,
|
| 119 |
+
"B-IBAN": 26,
|
| 120 |
+
"B-IMEI": 27,
|
| 121 |
+
"B-IPADDRESS": 28,
|
| 122 |
+
"B-JOBDEPARTMENT": 29,
|
| 123 |
+
"B-JOBTITLE": 30,
|
| 124 |
+
"B-LASTNAME": 31,
|
| 125 |
+
"B-LITECOINADDRESS": 32,
|
| 126 |
+
"B-MACADDRESS": 33,
|
| 127 |
+
"B-MASKEDNUMBER": 34,
|
| 128 |
+
"B-MIDDLENAME": 35,
|
| 129 |
+
"B-OCCUPATION": 36,
|
| 130 |
+
"B-ORDINALDIRECTION": 37,
|
| 131 |
+
"B-ORGANIZATION": 38,
|
| 132 |
+
"B-PASSWORD": 39,
|
| 133 |
+
"B-PHONE": 40,
|
| 134 |
+
"B-PIN": 41,
|
| 135 |
+
"B-PREFIX": 42,
|
| 136 |
+
"B-SECONDARYADDRESS": 43,
|
| 137 |
+
"B-SEX": 44,
|
| 138 |
+
"B-SSN": 45,
|
| 139 |
+
"B-STATE": 46,
|
| 140 |
+
"B-STREET": 47,
|
| 141 |
+
"B-TIME": 48,
|
| 142 |
+
"B-URL": 49,
|
| 143 |
+
"B-USERAGENT": 50,
|
| 144 |
+
"B-USERNAME": 51,
|
| 145 |
+
"B-VIN": 52,
|
| 146 |
+
"B-VRM": 53,
|
| 147 |
+
"B-ZIPCODE": 54,
|
| 148 |
+
"I-ACCOUNTNAME": 55,
|
| 149 |
+
"I-AGE": 56,
|
| 150 |
+
"I-AMOUNT": 57,
|
| 151 |
+
"I-CITY": 58,
|
| 152 |
+
"I-COUNTY": 59,
|
| 153 |
+
"I-CURRENCY": 60,
|
| 154 |
+
"I-CURRENCYNAME": 61,
|
| 155 |
+
"I-DATE": 62,
|
| 156 |
+
"I-DATEOFBIRTH": 63,
|
| 157 |
+
"I-EYECOLOR": 64,
|
| 158 |
+
"I-GENDER": 65,
|
| 159 |
+
"I-HEIGHT": 66,
|
| 160 |
+
"I-JOBTITLE": 67,
|
| 161 |
+
"I-ORGANIZATION": 68,
|
| 162 |
+
"I-PHONE": 69,
|
| 163 |
+
"I-SECONDARYADDRESS": 70,
|
| 164 |
+
"I-SSN": 71,
|
| 165 |
+
"I-STATE": 72,
|
| 166 |
+
"I-STREET": 73,
|
| 167 |
+
"I-TIME": 74,
|
| 168 |
+
"I-USERAGENT": 75,
|
| 169 |
+
"O": 0
|
| 170 |
+
},
|
| 171 |
+
"layer_types": [
|
| 172 |
+
"full_attention",
|
| 173 |
+
"full_attention",
|
| 174 |
+
"full_attention",
|
| 175 |
+
"full_attention",
|
| 176 |
+
"full_attention",
|
| 177 |
+
"full_attention",
|
| 178 |
+
"full_attention",
|
| 179 |
+
"full_attention",
|
| 180 |
+
"full_attention",
|
| 181 |
+
"full_attention",
|
| 182 |
+
"full_attention",
|
| 183 |
+
"full_attention",
|
| 184 |
+
"full_attention",
|
| 185 |
+
"full_attention",
|
| 186 |
+
"full_attention",
|
| 187 |
+
"full_attention",
|
| 188 |
+
"full_attention",
|
| 189 |
+
"full_attention",
|
| 190 |
+
"full_attention",
|
| 191 |
+
"full_attention",
|
| 192 |
+
"full_attention",
|
| 193 |
+
"full_attention",
|
| 194 |
+
"full_attention",
|
| 195 |
+
"full_attention",
|
| 196 |
+
"full_attention",
|
| 197 |
+
"full_attention",
|
| 198 |
+
"full_attention",
|
| 199 |
+
"full_attention"
|
| 200 |
+
],
|
| 201 |
+
"max_position_embeddings": 32768,
|
| 202 |
+
"max_window_layers": 28,
|
| 203 |
+
"model_type": "qwen3",
|
| 204 |
+
"num_attention_heads": 16,
|
| 205 |
+
"num_hidden_layers": 28,
|
| 206 |
+
"num_key_value_heads": 8,
|
| 207 |
+
"pad_token_id": 151643,
|
| 208 |
+
"rms_norm_eps": 1e-06,
|
| 209 |
+
"rope_parameters": {
|
| 210 |
+
"rope_theta": 1000000,
|
| 211 |
+
"rope_type": "default"
|
| 212 |
+
},
|
| 213 |
+
"sliding_window": null,
|
| 214 |
+
"tie_word_embeddings": true,
|
| 215 |
+
"transformers_version": "5.5.3",
|
| 216 |
+
"use_cache": false,
|
| 217 |
+
"use_sliding_window": false,
|
| 218 |
+
"vocab_size": 151669
|
| 219 |
+
}
|
eval_results.json
ADDED
|
@@ -0,0 +1,13 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"epoch": 3.0,
|
| 3 |
+
"eval_accuracy": 0.888999548715395,
|
| 4 |
+
"eval_f1": 0.7939432989690722,
|
| 5 |
+
"eval_loss": 0.5661426782608032,
|
| 6 |
+
"eval_macro_f1": 0.3903729497661312,
|
| 7 |
+
"eval_precision": 0.7801202912314024,
|
| 8 |
+
"eval_recall": 0.8082650049196458,
|
| 9 |
+
"eval_runtime": 2.8617,
|
| 10 |
+
"eval_samples_per_second": 535.703,
|
| 11 |
+
"eval_steps_per_second": 16.773,
|
| 12 |
+
"eval_weighted_f1": 0.8033758003197404
|
| 13 |
+
}
|
model.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:42ada723fc280903b178b9d444aa75c7ddc61667d8562f246fc90c1c8eb0ebe0
|
| 3 |
+
size 1191744248
|
test_results.json
ADDED
|
@@ -0,0 +1,12 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"test_accuracy": 0.8919662851455586,
|
| 3 |
+
"test_f1": 0.7929066580686763,
|
| 4 |
+
"test_loss": 0.5452415943145752,
|
| 5 |
+
"test_macro_f1": 0.40795926781161085,
|
| 6 |
+
"test_precision": 0.7832346009299955,
|
| 7 |
+
"test_recall": 0.8028205797858449,
|
| 8 |
+
"test_runtime": 2.8812,
|
| 9 |
+
"test_samples_per_second": 532.077,
|
| 10 |
+
"test_steps_per_second": 16.66,
|
| 11 |
+
"test_weighted_f1": 0.803044059461668
|
| 12 |
+
}
|
tokenizer.json
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:d997ad7d37f091d0cbef52058bf97af16e15b7bc06ccbdddd9b98bd9cf4deed8
|
| 3 |
+
size 11423799
|
tokenizer_config.json
ADDED
|
@@ -0,0 +1,14 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"add_prefix_space": false,
|
| 3 |
+
"backend": "tokenizers",
|
| 4 |
+
"bos_token": null,
|
| 5 |
+
"clean_up_tokenization_spaces": false,
|
| 6 |
+
"eos_token": "<|im_end|>",
|
| 7 |
+
"errors": "replace",
|
| 8 |
+
"is_local": false,
|
| 9 |
+
"model_max_length": 131072,
|
| 10 |
+
"pad_token": "<|endoftext|>",
|
| 11 |
+
"split_special_tokens": false,
|
| 12 |
+
"tokenizer_class": "Qwen2Tokenizer",
|
| 13 |
+
"unk_token": null
|
| 14 |
+
}
|
train_results.json
ADDED
|
@@ -0,0 +1,8 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"epoch": 3.0,
|
| 3 |
+
"total_flos": 5481934918516736.0,
|
| 4 |
+
"train_loss": 3.2885495954089694,
|
| 5 |
+
"train_runtime": 335.4941,
|
| 6 |
+
"train_samples_per_second": 109.683,
|
| 7 |
+
"train_steps_per_second": 1.717
|
| 8 |
+
}
|