Add compliance-classification

README.md

@@ -16,17 +16,18 @@ You can install it with `pip install adaptive-classifier`.
 ## Model Details
 
 - Base Model: answerdotai/ModernBERT-base
-- Number of Classes: 4
-- Total Examples: 352
+- Number of Classes: 5
+- Total Examples: 403
 - Embedding Dimension: 768
 
 ## Class Distribution
 
 ```
-compliant: 88 examples (25.0%)
-needs_review: 88 examples (25.0%)
-non_compliant: 88 examples (25.0%)
-unclear: 88 examples (25.0%)
+gdpr: 81 examples (20.1%)
+hipaa: 81 examples (20.1%)
+other: 80 examples (19.9%)
+pci: 80 examples (19.9%)
+sox: 81 examples (20.1%)
 ```
 
 ## Usage
@@ -50,7 +51,7 @@ classifier.add_examples(texts, labels)
 
 ## Training Details
 
-- Training Steps: 11
+- Training Steps: 13
 - Examples per Class: See distribution above
 - Prototype Memory: Active
 - Neural Adaptation: Active

config.json

@@ -32,17 +32,19 @@
   },
   "embedding_dim": 768,
   "id_to_label": {
-    "0": "unclear",
-    "1": "compliant",
-    "2": "needs_review",
-    "3": "non_compliant"
+    "0": "hipaa",
+    "1": "other",
+    "2": "pci",
+    "3": "gdpr",
+    "4": "sox"
   },
   "label_to_id": {
-    "compliant": 1,
-    "needs_review": 2,
-    "non_compliant": 3,
-    "unclear": 0
+    "gdpr": 3,
+    "hipaa": 0,
+    "other": 1,
+    "pci": 2,
+    "sox": 4
   },
   "model_name": "answerdotai/ModernBERT-base",
-  "train_steps": 11
+  "train_steps": 13
 }

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:c4ae42f6d568a10be53194e84b5f0c095b594fc70719e2c7e2bd9f9e88e8dd43
-size 3562912
+oid sha256:addd4ecc7c58937a375dc6c7591096f6fad2ca93abfe2119102b8473f3dabd84
+size 3567580