Upload Italian PII detection model OpenMed-PII-Italian-ModernMed-Large-395M-v1

README.md

@@ -0,0 +1,305 @@
+---
+language:
+  - it
+license: apache-2.0
+base_model: answerdotai/ModernBERT-large
+tags:
+  - token-classification
+  - ner
+  - pii
+  - pii-detection
+  - de-identification
+  - privacy
+  - healthcare
+  - medical
+  - clinical
+  - phi
+  - italian
+  - pytorch
+  - transformers
+  - openmed
+pipeline_tag: token-classification
+library_name: transformers
+metrics:
+  - f1
+  - precision
+  - recall
+model-index:
+  - name: OpenMed-PII-Italian-ModernMed-Large-395M-v1
+    results:
+      - task:
+          type: token-classification
+          name: Named Entity Recognition
+        dataset:
+          name: AI4Privacy (Italian subset)
+          type: ai4privacy/pii-masking-400k
+          split: test
+        metrics:
+          - type: f1
+            value: 0.9564
+            name: F1 (micro)
+          - type: precision
+            value: 0.9541
+            name: Precision
+          - type: recall
+            value: 0.9587
+            name: Recall
+widget:
+  - text: "Dr. Marco Rossi (Codice Fiscale: RSSMRC85C15H501Z) può essere contattato a marco.rossi@ospedale.it o al +39 333 123 4567. Abita in Via Roma 25, 00184 Roma."
+    example_title: Clinical Note with PII (Italian)
+---
+
+# OpenMed-PII-Italian-ModernMed-Large-395M-v1
+
+**Italian PII Detection Model** | 395M Parameters | Open Source
+
+[![F1 Score](https://img.shields.io/badge/F1-95.64%25-brightgreen)]() [![Precision](https://img.shields.io/badge/Precision-95.41%25-blue)]() [![Recall](https://img.shields.io/badge/Recall-95.87%25-orange)]()
+
+## Model Description
+
+**OpenMed-PII-Italian-ModernMed-Large-395M-v1** is a transformer-based token classification model fine-tuned for **Personally Identifiable Information (PII) detection in Italian text**. This model identifies and classifies **54 types of sensitive information** including names, addresses, social security numbers, medical record numbers, and more.
+
+### Key Features
+
+- **Italian-Optimized**: Specifically trained on Italian text for optimal performance
+- **High Accuracy**: Achieves strong F1 scores across diverse PII categories
+- **Comprehensive Coverage**: Detects 55+ entity types spanning personal, financial, medical, and contact information
+- **Privacy-Focused**: Designed for de-identification and compliance with GDPR and other privacy regulations
+- **Production-Ready**: Optimized for real-world text processing pipelines
+
+## Performance
+
+Evaluated on the Italian subset of AI4Privacy dataset:
+
+| Metric | Score |
+|:---|:---:|
+| **Micro F1** | **0.9564** |
+| Precision | 0.9541 |
+| Recall | 0.9587 |
+| Macro F1 | 0.9491 |
+| Weighted F1 | 0.9563 |
+| Accuracy | 0.9940 |
+
+### Top 10 Italian PII Models
+
+| Rank | Model | F1 | Precision | Recall |
+|:---:|:---|:---:|:---:|:---:|
+| 1 | [OpenMed-PII-Italian-SuperClinical-Large-434M-v1](https://huggingface.co/OpenMed/OpenMed-PII-Italian-SuperClinical-Large-434M-v1) | 0.9728 | 0.9707 | 0.9750 |
+| 2 | [OpenMed-PII-Italian-EuroMed-210M-v1](https://huggingface.co/OpenMed/OpenMed-PII-Italian-EuroMed-210M-v1) | 0.9685 | 0.9663 | 0.9707 |
+| 3 | [OpenMed-PII-Italian-ClinicalBGE-568M-v1](https://huggingface.co/OpenMed/OpenMed-PII-Italian-ClinicalBGE-568M-v1) | 0.9678 | 0.9653 | 0.9703 |
+| 4 | [OpenMed-PII-Italian-SnowflakeMed-Large-568M-v1](https://huggingface.co/OpenMed/OpenMed-PII-Italian-SnowflakeMed-Large-568M-v1) | 0.9678 | 0.9653 | 0.9702 |
+| 5 | [OpenMed-PII-Italian-BigMed-Large-560M-v1](https://huggingface.co/OpenMed/OpenMed-PII-Italian-BigMed-Large-560M-v1) | 0.9671 | 0.9645 | 0.9697 |
+| 6 | [OpenMed-PII-Italian-SuperMedical-Large-355M-v1](https://huggingface.co/OpenMed/OpenMed-PII-Italian-SuperMedical-Large-355M-v1) | 0.9663 | 0.9640 | 0.9686 |
+| 7 | [OpenMed-PII-Italian-mClinicalE5-Large-560M-v1](https://huggingface.co/OpenMed/OpenMed-PII-Italian-mClinicalE5-Large-560M-v1) | 0.9659 | 0.9633 | 0.9684 |
+| 8 | [OpenMed-PII-Italian-NomicMed-Large-395M-v1](https://huggingface.co/OpenMed/OpenMed-PII-Italian-NomicMed-Large-395M-v1) | 0.9656 | 0.9631 | 0.9682 |
+| 9 | [OpenMed-PII-Italian-ClinicalBGE-Large-335M-v1](https://huggingface.co/OpenMed/OpenMed-PII-Italian-ClinicalBGE-Large-335M-v1) | 0.9605 | 0.9575 | 0.9635 |
+| 10 | [OpenMed-PII-Italian-SuperClinical-Base-184M-v1](https://huggingface.co/OpenMed/OpenMed-PII-Italian-SuperClinical-Base-184M-v1) | 0.9596 | 0.9573 | 0.9620 |
+
+## Supported Entity Types
+
+This model detects **54 PII entity types** organized into categories:
+
+<details>
+<summary><strong>Identifiers</strong> (22 types)</summary>
+
+| Entity | Description |
+|:---|:---|
+| `ACCOUNTNAME` | Accountname |
+| `BANKACCOUNT` | Bankaccount |
+| `BIC` | Bic |
+| `BITCOINADDRESS` | Bitcoinaddress |
+| `CREDITCARD` | Creditcard |
+| `CREDITCARDISSUER` | Creditcardissuer |
+| `CVV` | Cvv |
+| `ETHEREUMADDRESS` | Ethereumaddress |
+| `IBAN` | Iban |
+| `IMEI` | Imei |
+| ... | *and 12 more* |
+
+</details>
+
+<details>
+<summary><strong>Personal Info</strong> (11 types)</summary>
+
+| Entity | Description |
+|:---|:---|
+| `AGE` | Age |
+| `DATEOFBIRTH` | Dateofbirth |
+| `EYECOLOR` | Eyecolor |
+| `FIRSTNAME` | Firstname |
+| `GENDER` | Gender |
+| `HEIGHT` | Height |
+| `LASTNAME` | Lastname |
+| `MIDDLENAME` | Middlename |
+| `OCCUPATION` | Occupation |
+| `PREFIX` | Prefix |
+| ... | *and 1 more* |
+
+</details>
+
+<details>
+<summary><strong>Contact Info</strong> (2 types)</summary>
+
+| Entity | Description |
+|:---|:---|
+| `EMAIL` | Email |
+| `PHONE` | Phone |
+
+</details>
+
+<details>
+<summary><strong>Location</strong> (9 types)</summary>
+
+| Entity | Description |
+|:---|:---|
+| `BUILDINGNUMBER` | Buildingnumber |
+| `CITY` | City |
+| `COUNTY` | County |
+| `GPSCOORDINATES` | Gpscoordinates |
+| `ORDINALDIRECTION` | Ordinaldirection |
+| `SECONDARYADDRESS` | Secondaryaddress |
+| `STATE` | State |
+| `STREET` | Street |
+| `ZIPCODE` | Zipcode |
+
+</details>
+
+<details>
+<summary><strong>Organization</strong> (3 types)</summary>
+
+| Entity | Description |
+|:---|:---|
+| `JOBDEPARTMENT` | Jobdepartment |
+| `JOBTITLE` | Jobtitle |
+| `ORGANIZATION` | Organization |
+
+</details>
+
+<details>
+<summary><strong>Financial</strong> (5 types)</summary>
+
+| Entity | Description |
+|:---|:---|
+| `AMOUNT` | Amount |
+| `CURRENCY` | Currency |
+| `CURRENCYCODE` | Currencycode |
+| `CURRENCYNAME` | Currencyname |
+| `CURRENCYSYMBOL` | Currencysymbol |
+
+</details>
+
+<details>
+<summary><strong>Temporal</strong> (2 types)</summary>
+
+| Entity | Description |
+|:---|:---|
+| `DATE` | Date |
+| `TIME` | Time |
+
+</details>
+
+## Usage
+
+### Quick Start
+
+```python
+from transformers import pipeline
+
+# Load the PII detection pipeline
+ner = pipeline("ner", model="OpenMed/OpenMed-PII-Italian-ModernMed-Large-395M-v1", aggregation_strategy="simple")
+
+text = """
+Paziente Marco Bianchi (nato il 15/03/1985, CF: BNCMRC85C15H501Z) è stato visitato oggi.
+Contatto: marco.bianchi@email.it, Telefono: +39 333 123 4567.
+Indirizzo: Via Garibaldi 42, 20121 Milano.
+"""
+
+entities = ner(text)
+for entity in entities:
+    print(f"{entity['entity_group']}: {entity['word']} (score: {entity['score']:.3f})")
+```
+
+### De-identification Example
+
+```python
+def redact_pii(text, entities, placeholder='[REDACTED]'):
+    """Replace detected PII with placeholders."""
+    # Sort entities by start position (descending) to preserve offsets
+    sorted_entities = sorted(entities, key=lambda x: x['start'], reverse=True)
+    redacted = text
+    for ent in sorted_entities:
+        redacted = redacted[:ent['start']] + f"[{ent['entity_group']}]" + redacted[ent['end']:]
+    return redacted
+
+# Apply de-identification
+redacted_text = redact_pii(text, entities)
+print(redacted_text)
+```
+
+### Batch Processing
+
+```python
+from transformers import AutoModelForTokenClassification, AutoTokenizer
+import torch
+
+model_name = "OpenMed/OpenMed-PII-Italian-ModernMed-Large-395M-v1"
+model = AutoModelForTokenClassification.from_pretrained(model_name)
+tokenizer = AutoTokenizer.from_pretrained(model_name)
+
+texts = [
+    "Paziente Marco Bianchi (nato il 15/03/1985, CF: BNCMRC85C15H501Z) è stato visitato oggi.",
+    "Contatto: marco.bianchi@email.it, Telefono: +39 333 123 4567.",
+]
+
+inputs = tokenizer(texts, return_tensors='pt', padding=True, truncation=True)
+with torch.no_grad():
+    outputs = model(**inputs)
+    predictions = torch.argmax(outputs.logits, dim=-1)
+```
+
+## Training Details
+
+### Dataset
+
+- **Source**: [AI4Privacy PII Masking 400k](https://huggingface.co/datasets/ai4privacy/pii-masking-400k) (Italian subset)
+- **Format**: BIO-tagged token classification
+- **Labels**: 109 total (54 entity types × 2 BIO tags + O)
+
+### Training Configuration
+
+- **Max Sequence Length**: 512 tokens
+- **Epochs**: 3
+- **Framework**: Hugging Face Transformers + Trainer API
+
+## Intended Use & Limitations
+
+### Intended Use
+
+- **De-identification**: Automated redaction of PII in Italian clinical notes, medical records, and documents
+- **Compliance**: Supporting GDPR, and other privacy regulation compliance
+- **Data Preprocessing**: Preparing datasets for research by removing sensitive information
+- **Audit Support**: Identifying PII in document collections
+
+### Limitations
+
+**Important**: This model is intended as an **assistive tool**, not a replacement for human review.
+
+- **False Negatives**: Some PII may not be detected; always verify critical applications
+- **Context Sensitivity**: Performance may vary with domain-specific terminology
+- **Language**: Optimized for Italian text; may not perform well on other languages
+
+## Citation
+
+```bibtex
+@misc{openmed-pii-2026,
+  title = {OpenMed-PII-Italian-ModernMed-Large-395M-v1: Italian PII Detection Model},
+  author = {OpenMed Science},
+  year = {2026},
+  publisher = {Hugging Face},
+  url = {https://huggingface.co/OpenMed/OpenMed-PII-Italian-ModernMed-Large-395M-v1}
+}
+```
+
+## Links
+
+- **Organization**: [OpenMed](https://huggingface.co/OpenMed)

all_results.json

@@ -0,0 +1,28 @@
+{
+    "epoch": 3.0,
+    "eval_accuracy": 0.994844131159147,
+    "eval_f1": 0.9639322033898305,
+    "eval_loss": 0.014384633861482143,
+    "eval_macro_f1": 0.9541869985075359,
+    "eval_precision": 0.9613902224626412,
+    "eval_recall": 0.9664876622935219,
+    "eval_runtime": 7.6622,
+    "eval_samples_per_second": 648.903,
+    "eval_steps_per_second": 20.36,
+    "eval_weighted_f1": 0.9641413661433497,
+    "test_accuracy": 0.9939838776845737,
+    "test_f1": 0.9563980415508799,
+    "test_loss": 0.01686817966401577,
+    "test_macro_f1": 0.9490995419020036,
+    "test_precision": 0.9540624381228962,
+    "test_recall": 0.9587451084433243,
+    "test_runtime": 8.6689,
+    "test_samples_per_second": 584.735,
+    "test_steps_per_second": 18.341,
+    "test_weighted_f1": 0.9562803095255936,
+    "total_flos": 1.881641387556864e+16,
+    "train_loss": 0.08123073937216153,
+    "train_runtime": 961.2851,
+    "train_samples_per_second": 127.779,
+    "train_steps_per_second": 1.997
+}

classification_report.txt

@@ -0,0 +1,64 @@
+Classification Report for Italian PII Detection
+Model: answerdotai/ModernBERT-large
+============================================================
+
+                  precision    recall  f1-score   support
+
+     ACCOUNTNAME       0.99      1.00      1.00       282
+             AGE       0.99      0.99      0.99       338
+          AMOUNT       0.98      0.99      0.99       116
+     BANKACCOUNT       1.00      1.00      1.00       306
+             BIC       1.00      1.00      1.00        77
+  BITCOINADDRESS       0.93      0.99      0.96       273
+  BUILDINGNUMBER       0.93      0.91      0.92       346
+            CITY       0.96      0.91      0.93       280
+          COUNTY       0.98      1.00      0.99       327
+      CREDITCARD       0.84      0.89      0.86       302
+CREDITCARDISSUER       1.00      1.00      1.00       146
+        CURRENCY       0.68      0.82      0.74       187
+    CURRENCYCODE       0.98      0.96      0.97        85
+    CURRENCYNAME       0.46      0.36      0.40        97
+  CURRENCYSYMBOL       0.98      0.98      0.98       308
+             CVV       0.96      0.99      0.97        97
+            DATE       0.73      0.85      0.79       423
+     DATEOFBIRTH       0.73      0.65      0.68       327
+           EMAIL       1.00      1.00      1.00       423
+ ETHEREUMADDRESS       1.00      1.00      1.00       168
+        EYECOLOR       1.00      0.98      0.99       108
+       FIRSTNAME       0.97      0.97      0.97      1623
+          GENDER       0.99      0.99      0.99       302
+  GPSCOORDINATES       1.00      1.00      1.00       223
+          HEIGHT       0.99      0.99      0.99       126
+            IBAN       0.99      1.00      1.00       230
+            IMEI       1.00      1.00      1.00       215
+       IPADDRESS       1.00      1.00      1.00       783
+   JOBDEPARTMENT       0.99      0.98      0.98       327
+        JOBTITLE       0.99      1.00      0.99       279
+        LASTNAME       0.95      0.94      0.95       441
+ LITECOINADDRESS       0.93      0.76      0.83        83
+      MACADDRESS       0.99      0.98      0.99       114
+    MASKEDNUMBER       0.82      0.76      0.79       209
+      MIDDLENAME       0.90      0.96      0.93       310
+      OCCUPATION       1.00      0.99      1.00       323
+ORDINALDIRECTION       1.00      1.00      1.00       152
+    ORGANIZATION       0.98      0.99      0.99       272
+        PASSWORD       0.99      0.98      0.98       286
+           PHONE       0.99      1.00      0.99       303
+             PIN       0.90      0.90      0.90        72
+          PREFIX       0.98      1.00      0.99       298
+SECONDARYADDRESS       0.99      1.00      1.00       316
+             SEX       1.00      1.00      1.00       338
+             SSN       1.00      1.00      1.00       259
+           STATE       0.98      0.99      0.98       294
+          STREET       0.96      0.97      0.97       332
+            TIME       0.99      0.99      0.99       296
+             URL       1.00      1.00      1.00       244
+       USERAGENT       0.99      1.00      0.99       233
+        USERNAME       0.99      0.98      0.99       332
+             VIN       1.00      1.00      1.00        84
+             VRM       1.00      1.00      1.00        98
+         ZIPCODE       0.92      0.92      0.92       264
+
+       micro avg       0.95      0.96      0.96     15077
+       macro avg       0.95      0.95      0.95     15077
+    weighted avg       0.95      0.96      0.96     15077

config.json

@@ -0,0 +1,201 @@
+{
+  "architectures": [
+    "ModernBertForTokenClassification"
+  ],
+  "attention_bias": false,
+  "attention_dropout": 0.0,
+  "bos_token_id": null,
+  "classifier_activation": "gelu",
+  "classifier_bias": false,
+  "classifier_dropout": 0.0,
+  "classifier_pooling": "mean",
+  "cls_token_id": 50281,
+  "decoder_bias": true,
+  "deterministic_flash_attn": false,
+  "dtype": "float32",
+  "embedding_dropout": 0.0,
+  "eos_token_id": null,
+  "global_attn_every_n_layers": 3,
+  "global_rope_theta": 160000.0,
+  "gradient_checkpointing": false,
+  "hidden_activation": "gelu",
+  "hidden_size": 1024,
+  "id2label": {
+    "0": "O",
+    "1": "B-ACCOUNTNAME",
+    "2": "B-AGE",
+    "3": "B-AMOUNT",
+    "4": "B-BANKACCOUNT",
+    "5": "B-BIC",
+    "6": "B-BITCOINADDRESS",
+    "7": "B-BUILDINGNUMBER",
+    "8": "B-CITY",
+    "9": "B-COUNTY",
+    "10": "B-CREDITCARD",
+    "11": "B-CREDITCARDISSUER",
+    "12": "B-CURRENCY",
+    "13": "B-CURRENCYCODE",
+    "14": "B-CURRENCYNAME",
+    "15": "B-CURRENCYSYMBOL",
+    "16": "B-CVV",
+    "17": "B-DATE",
+    "18": "B-DATEOFBIRTH",
+    "19": "B-EMAIL",
+    "20": "B-ETHEREUMADDRESS",
+    "21": "B-EYECOLOR",
+    "22": "B-FIRSTNAME",
+    "23": "B-GENDER",
+    "24": "B-GPSCOORDINATES",
+    "25": "B-HEIGHT",
+    "26": "B-IBAN",
+    "27": "B-IMEI",
+    "28": "B-IPADDRESS",
+    "29": "B-JOBDEPARTMENT",
+    "30": "B-JOBTITLE",
+    "31": "B-LASTNAME",
+    "32": "B-LITECOINADDRESS",
+    "33": "B-MACADDRESS",
+    "34": "B-MASKEDNUMBER",
+    "35": "B-MIDDLENAME",
+    "36": "B-OCCUPATION",
+    "37": "B-ORDINALDIRECTION",
+    "38": "B-ORGANIZATION",
+    "39": "B-PASSWORD",
+    "40": "B-PHONE",
+    "41": "B-PIN",
+    "42": "B-PREFIX",
+    "43": "B-SECONDARYADDRESS",
+    "44": "B-SEX",
+    "45": "B-SSN",
+    "46": "B-STATE",
+    "47": "B-STREET",
+    "48": "B-TIME",
+    "49": "B-URL",
+    "50": "B-USERAGENT",
+    "51": "B-USERNAME",
+    "52": "B-VIN",
+    "53": "B-VRM",
+    "54": "B-ZIPCODE",
+    "55": "I-ACCOUNTNAME",
+    "56": "I-AGE",
+    "57": "I-AMOUNT",
+    "58": "I-CITY",
+    "59": "I-COUNTY",
+    "60": "I-CURRENCY",
+    "61": "I-CURRENCYNAME",
+    "62": "I-DATE",
+    "63": "I-DATEOFBIRTH",
+    "64": "I-EYECOLOR",
+    "65": "I-GENDER",
+    "66": "I-HEIGHT",
+    "67": "I-JOBTITLE",
+    "68": "I-ORGANIZATION",
+    "69": "I-PHONE",
+    "70": "I-SECONDARYADDRESS",
+    "71": "I-SSN",
+    "72": "I-STATE",
+    "73": "I-STREET",
+    "74": "I-TIME",
+    "75": "I-USERAGENT"
+  },
+  "initializer_cutoff_factor": 2.0,
+  "initializer_range": 0.02,
+  "intermediate_size": 2624,
+  "label2id": {
+    "B-ACCOUNTNAME": 1,
+    "B-AGE": 2,
+    "B-AMOUNT": 3,
+    "B-BANKACCOUNT": 4,
+    "B-BIC": 5,
+    "B-BITCOINADDRESS": 6,
+    "B-BUILDINGNUMBER": 7,
+    "B-CITY": 8,
+    "B-COUNTY": 9,
+    "B-CREDITCARD": 10,
+    "B-CREDITCARDISSUER": 11,
+    "B-CURRENCY": 12,
+    "B-CURRENCYCODE": 13,
+    "B-CURRENCYNAME": 14,
+    "B-CURRENCYSYMBOL": 15,
+    "B-CVV": 16,
+    "B-DATE": 17,
+    "B-DATEOFBIRTH": 18,
+    "B-EMAIL": 19,
+    "B-ETHEREUMADDRESS": 20,
+    "B-EYECOLOR": 21,
+    "B-FIRSTNAME": 22,
+    "B-GENDER": 23,
+    "B-GPSCOORDINATES": 24,
+    "B-HEIGHT": 25,
+    "B-IBAN": 26,
+    "B-IMEI": 27,
+    "B-IPADDRESS": 28,
+    "B-JOBDEPARTMENT": 29,
+    "B-JOBTITLE": 30,
+    "B-LASTNAME": 31,
+    "B-LITECOINADDRESS": 32,
+    "B-MACADDRESS": 33,
+    "B-MASKEDNUMBER": 34,
+    "B-MIDDLENAME": 35,
+    "B-OCCUPATION": 36,
+    "B-ORDINALDIRECTION": 37,
+    "B-ORGANIZATION": 38,
+    "B-PASSWORD": 39,
+    "B-PHONE": 40,
+    "B-PIN": 41,
+    "B-PREFIX": 42,
+    "B-SECONDARYADDRESS": 43,
+    "B-SEX": 44,
+    "B-SSN": 45,
+    "B-STATE": 46,
+    "B-STREET": 47,
+    "B-TIME": 48,
+    "B-URL": 49,
+    "B-USERAGENT": 50,
+    "B-USERNAME": 51,
+    "B-VIN": 52,
+    "B-VRM": 53,
+    "B-ZIPCODE": 54,
+    "I-ACCOUNTNAME": 55,
+    "I-AGE": 56,
+    "I-AMOUNT": 57,
+    "I-CITY": 58,
+    "I-COUNTY": 59,
+    "I-CURRENCY": 60,
+    "I-CURRENCYNAME": 61,
+    "I-DATE": 62,
+    "I-DATEOFBIRTH": 63,
+    "I-EYECOLOR": 64,
+    "I-GENDER": 65,
+    "I-HEIGHT": 66,
+    "I-JOBTITLE": 67,
+    "I-ORGANIZATION": 68,
+    "I-PHONE": 69,
+    "I-SECONDARYADDRESS": 70,
+    "I-SSN": 71,
+    "I-STATE": 72,
+    "I-STREET": 73,
+    "I-TIME": 74,
+    "I-USERAGENT": 75,
+    "O": 0
+  },
+  "layer_norm_eps": 1e-05,
+  "local_attention": 128,
+  "local_rope_theta": 10000.0,
+  "max_position_embeddings": 8192,
+  "mlp_bias": false,
+  "mlp_dropout": 0.0,
+  "model_type": "modernbert",
+  "norm_bias": false,
+  "norm_eps": 1e-05,
+  "num_attention_heads": 16,
+  "num_hidden_layers": 28,
+  "pad_token_id": 50283,
+  "position_embedding_type": "absolute",
+  "repad_logits_with_grad": false,
+  "sep_token_id": 50282,
+  "sparse_pred_ignore_index": -100,
+  "sparse_prediction": false,
+  "transformers_version": "4.57.3",
+  "vocab_size": 50368
+}

eval_results.json

@@ -0,0 +1,13 @@
+{
+    "epoch": 3.0,
+    "eval_accuracy": 0.994844131159147,
+    "eval_f1": 0.9639322033898305,
+    "eval_loss": 0.014384633861482143,
+    "eval_macro_f1": 0.9541869985075359,
+    "eval_precision": 0.9613902224626412,
+    "eval_recall": 0.9664876622935219,
+    "eval_runtime": 7.6622,
+    "eval_samples_per_second": 648.903,
+    "eval_steps_per_second": 20.36,
+    "eval_weighted_f1": 0.9641413661433497
+}

model.safetensors

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

special_tokens_map.json

@@ -0,0 +1,37 @@
+{
+  "cls_token": {
+    "content": "[CLS]",
+    "lstrip": false,
+    "normalized": false,
+    "rstrip": false,
+    "single_word": false
+  },
+  "mask_token": {
+    "content": "[MASK]",
+    "lstrip": true,
+    "normalized": false,
+    "rstrip": false,
+    "single_word": false
+  },
+  "pad_token": {
+    "content": "[PAD]",
+    "lstrip": false,
+    "normalized": false,
+    "rstrip": false,
+    "single_word": false
+  },
+  "sep_token": {
+    "content": "[SEP]",
+    "lstrip": false,
+    "normalized": false,
+    "rstrip": false,
+    "single_word": false
+  },
+  "unk_token": {
+    "content": "[UNK]",
+    "lstrip": false,
+    "normalized": false,
+    "rstrip": false,
+    "single_word": false
+  }
+}

test_results.json

@@ -0,0 +1,12 @@
+{
+    "test_accuracy": 0.9939838776845737,
+    "test_f1": 0.9563980415508799,
+    "test_loss": 0.01686817966401577,
+    "test_macro_f1": 0.9490995419020036,
+    "test_precision": 0.9540624381228962,
+    "test_recall": 0.9587451084433243,
+    "test_runtime": 8.6689,
+    "test_samples_per_second": 584.735,
+    "test_steps_per_second": 18.341,
+    "test_weighted_f1": 0.9562803095255936
+}

tokenizer_config.json

@@ -0,0 +1,945 @@
+{
+  "added_tokens_decoder": {
+    "0": {
+      "content": "|||IP_ADDRESS|||",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "1": {
+      "content": "<|padding|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "50254": {
+      "content": "                        ",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50255": {
+      "content": "                       ",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50256": {
+      "content": "                      ",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50257": {
+      "content": "                     ",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50258": {
+      "content": "                    ",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50259": {
+      "content": "                   ",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50260": {
+      "content": "                  ",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50261": {
+      "content": "                 ",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50262": {
+      "content": "                ",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50263": {
+      "content": "               ",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50264": {
+      "content": "              ",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50265": {
+      "content": "             ",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50266": {
+      "content": "            ",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50267": {
+      "content": "           ",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50268": {
+      "content": "          ",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50269": {
+      "content": "         ",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50270": {
+      "content": "        ",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50271": {
+      "content": "       ",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50272": {
+      "content": "      ",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50273": {
+      "content": "     ",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50274": {
+      "content": "    ",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50275": {
+      "content": "   ",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50276": {
+      "content": "  ",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50277": {
+      "content": "|||EMAIL_ADDRESS|||",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50278": {
+      "content": "|||PHONE_NUMBER|||",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50279": {
+      "content": "<|endoftext|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "50280": {
+      "content": "[UNK]",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "50281": {
+      "content": "[CLS]",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "50282": {
+      "content": "[SEP]",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "50283": {
+      "content": "[PAD]",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "50284": {
+      "content": "[MASK]",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "50285": {
+      "content": "[unused0]",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50286": {
+      "content": "[unused1]",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50287": {
+      "content": "[unused2]",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50288": {
+      "content": "[unused3]",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50289": {
+      "content": "[unused4]",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50290": {
+      "content": "[unused5]",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50291": {
+      "content": "[unused6]",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50292": {
+      "content": "[unused7]",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50293": {
+      "content": "[unused8]",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50294": {
+      "content": "[unused9]",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50295": {
+      "content": "[unused10]",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50296": {
+      "content": "[unused11]",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50297": {
+      "content": "[unused12]",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50298": {
+      "content": "[unused13]",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50299": {
+      "content": "[unused14]",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50300": {
+      "content": "[unused15]",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50301": {
+      "content": "[unused16]",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50302": {
+      "content": "[unused17]",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50303": {
+      "content": "[unused18]",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50304": {
+      "content": "[unused19]",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50305": {
+      "content": "[unused20]",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50306": {
+      "content": "[unused21]",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50307": {
+      "content": "[unused22]",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50308": {
+      "content": "[unused23]",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50309": {
+      "content": "[unused24]",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50310": {
+      "content": "[unused25]",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50311": {
+      "content": "[unused26]",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50312": {
+      "content": "[unused27]",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50313": {
+      "content": "[unused28]",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50314": {
+      "content": "[unused29]",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50315": {
+      "content": "[unused30]",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50316": {
+      "content": "[unused31]",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50317": {
+      "content": "[unused32]",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50318": {
+      "content": "[unused33]",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50319": {
+      "content": "[unused34]",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50320": {
+      "content": "[unused35]",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50321": {
+      "content": "[unused36]",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50322": {
+      "content": "[unused37]",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50323": {
+      "content": "[unused38]",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50324": {
+      "content": "[unused39]",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50325": {
+      "content": "[unused40]",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50326": {
+      "content": "[unused41]",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50327": {
+      "content": "[unused42]",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50328": {
+      "content": "[unused43]",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50329": {
+      "content": "[unused44]",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50330": {
+      "content": "[unused45]",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50331": {
+      "content": "[unused46]",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50332": {
+      "content": "[unused47]",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50333": {
+      "content": "[unused48]",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50334": {
+      "content": "[unused49]",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50335": {
+      "content": "[unused50]",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50336": {
+      "content": "[unused51]",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50337": {
+      "content": "[unused52]",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50338": {
+      "content": "[unused53]",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50339": {
+      "content": "[unused54]",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50340": {
+      "content": "[unused55]",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50341": {
+      "content": "[unused56]",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50342": {
+      "content": "[unused57]",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50343": {
+      "content": "[unused58]",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50344": {
+      "content": "[unused59]",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50345": {
+      "content": "[unused60]",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50346": {
+      "content": "[unused61]",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50347": {
+      "content": "[unused62]",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50348": {
+      "content": "[unused63]",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50349": {
+      "content": "[unused64]",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50350": {
+      "content": "[unused65]",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50351": {
+      "content": "[unused66]",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50352": {
+      "content": "[unused67]",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50353": {
+      "content": "[unused68]",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50354": {
+      "content": "[unused69]",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50355": {
+      "content": "[unused70]",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50356": {
+      "content": "[unused71]",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50357": {
+      "content": "[unused72]",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50358": {
+      "content": "[unused73]",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50359": {
+      "content": "[unused74]",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50360": {
+      "content": "[unused75]",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50361": {
+      "content": "[unused76]",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50362": {
+      "content": "[unused77]",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50363": {
+      "content": "[unused78]",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50364": {
+      "content": "[unused79]",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50365": {
+      "content": "[unused80]",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50366": {
+      "content": "[unused81]",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50367": {
+      "content": "[unused82]",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    }
+  },
+  "clean_up_tokenization_spaces": true,
+  "cls_token": "[CLS]",
+  "extra_special_tokens": {},
+  "mask_token": "[MASK]",
+  "model_input_names": [
+    "input_ids",
+    "attention_mask"
+  ],
+  "model_max_length": 8192,
+  "pad_token": "[PAD]",
+  "sep_token": "[SEP]",
+  "tokenizer_class": "PreTrainedTokenizerFast",
+  "unk_token": "[UNK]"
+}

train_results.json

@@ -0,0 +1,8 @@
+{
+    "epoch": 3.0,
+    "total_flos": 1.881641387556864e+16,
+    "train_loss": 0.08123073937216153,
+    "train_runtime": 961.2851,
+    "train_samples_per_second": 127.779,
+    "train_steps_per_second": 1.997
+}