Add content-moderation

README.md

@@ -16,17 +16,16 @@ You can install it with `pip install adaptive-classifier`.
 ## Model Details
 
 - Base Model: answerdotai/ModernBERT-base
-- Number of Classes: 4
-- Total Examples: 376
+- Number of Classes: 3
+- Total Examples: 252
 - Embedding Dimension: 768
 
 ## Class Distribution
 
 ```
-appropriate: 96 examples (25.5%)
-inappropriate: 96 examples (25.5%)
-needs_review: 96 examples (25.5%)
-toxic: 88 examples (23.4%)
+appropriate: 86 examples (34.1%)
+inappropriate: 82 examples (32.5%)
+spam: 84 examples (33.3%)
 ```
 
 ## Usage
@@ -50,7 +49,7 @@ classifier.add_examples(texts, labels)
 
 ## Training Details
 
-- Training Steps: 12
+- Training Steps: 8
 - Examples per Class: See distribution above
 - Prototype Memory: Active
 - Neural Adaptation: Active

config.json

@@ -32,17 +32,15 @@
   },
   "embedding_dim": 768,
   "id_to_label": {
-    "0": "toxic",
-    "1": "inappropriate",
-    "2": "needs_review",
-    "3": "appropriate"
+    "0": "appropriate",
+    "1": "spam",
+    "2": "inappropriate"
   },
   "label_to_id": {
-    "appropriate": 3,
-    "inappropriate": 1,
-    "needs_review": 2,
-    "toxic": 0
+    "appropriate": 0,
+    "inappropriate": 2,
+    "spam": 1
   },
   "model_name": "answerdotai/ModernBERT-base",
-  "train_steps": 12
+  "train_steps": 8
 }

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:2d4a5ccfef03c9952c3a40c83b1d9d37a4271ef8a24f9b69ec549a81f096de37
-size 3562912
+oid sha256:135f1483ace80b2fe4f8f339df5e40104e934611e1043d814fe118283691aab7
+size 3558212