Add product-category

README.md

@@ -16,19 +16,17 @@ You can install it with `pip install adaptive-classifier`.
 ## Model Details
 
 - Base Model: answerdotai/ModernBERT-base
-- Number of Classes: 6
-- Total Examples: 480
+- Number of Classes: 4
+- Total Examples: 321
 - Embedding Dimension: 768
 
 ## Class Distribution
 
 ```
-consulting: 80 examples (16.7%)
-hardware: 80 examples (16.7%)
-other: 80 examples (16.7%)
-service: 80 examples (16.7%)
-software: 80 examples (16.7%)
-subscription: 80 examples (16.7%)
+books: 81 examples (25.2%)
+clothing: 80 examples (24.9%)
+electronics: 80 examples (24.9%)
+home_garden: 80 examples (24.9%)
 ```
 
 ## Usage
@@ -52,7 +50,7 @@ classifier.add_examples(texts, labels)
 
 ## Training Details
 
-- Training Steps: 15
+- Training Steps: 11
 - Examples per Class: See distribution above
 - Prototype Memory: Active
 - Neural Adaptation: Active

config.json

@@ -32,21 +32,17 @@
   },
   "embedding_dim": 768,
   "id_to_label": {
-    "0": "consulting",
-    "1": "hardware",
-    "2": "service",
-    "3": "subscription",
-    "4": "other",
-    "5": "software"
+    "0": "clothing",
+    "1": "electronics",
+    "2": "books",
+    "3": "home_garden"
   },
   "label_to_id": {
-    "consulting": 0,
-    "hardware": 1,
-    "other": 4,
-    "service": 2,
-    "software": 5,
-    "subscription": 3
+    "books": 2,
+    "clothing": 0,
+    "electronics": 1,
+    "home_garden": 3
   },
   "model_name": "answerdotai/ModernBERT-base",
-  "train_steps": 15
+  "train_steps": 11
 }

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:7d4abe3b812a0c4c11c7a191b65df7ec90b051a2da44b53bf57ecae2b153df57
-size 3572296
+oid sha256:f788910ec9d2c7015d51fb712584303132ab75229723075e94fd41eee36f2f77
+size 3562904