Add pii-detection

README.md

@@ -16,16 +16,15 @@ You can install it with `pip install adaptive-classifier`.
 ## Model Details
 
 - Base Model: answerdotai/ModernBERT-base
-- Number of Classes: 3
-- Total Examples: 264
+- Number of Classes: 2
+- Total Examples: 163
 - Embedding Dimension: 768
 
 ## Class Distribution
 
 ```
-contains_pii: 88 examples (33.3%)
-no_pii: 88 examples (33.3%)
-partial_pii: 88 examples (33.3%)
+contains_pii: 81 examples (49.7%)
+no_pii: 82 examples (50.3%)
 ```
 
 ## Usage
@@ -49,7 +48,7 @@ classifier.add_examples(texts, labels)
 
 ## Training Details
 
-- Training Steps: 9
+- Training Steps: 6
 - Examples per Class: See distribution above
 - Prototype Memory: Active
 - Neural Adaptation: Active

config.json

@@ -32,15 +32,13 @@
   },
   "embedding_dim": 768,
   "id_to_label": {
-    "0": "no_pii",
-    "1": "contains_pii",
-    "2": "partial_pii"
+    "0": "contains_pii",
+    "1": "no_pii"
   },
   "label_to_id": {
-    "contains_pii": 1,
-    "no_pii": 0,
-    "partial_pii": 2
+    "contains_pii": 0,
+    "no_pii": 1
   },
   "model_name": "answerdotai/ModernBERT-base",
-  "train_steps": 9
+  "train_steps": 6
 }

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:6626aefe432b5ff74d705814c5e3245e48d758c8bb4c5be090190baecfd09512
-size 3558212
+oid sha256:882cc21eedddbc1cf8868a363c2089c29c60937ef4e251421b9756f3d2f350ae
+size 3553512