Add expense-category

README.md

@@ -16,19 +16,18 @@ You can install it with `pip install adaptive-classifier`.
 ## Model Details
 
 - Base Model: answerdotai/ModernBERT-base
-- Number of Classes: 6
-- Total Examples: 30
+- Number of Classes: 5
+- Total Examples: 401
 - Embedding Dimension: 768
 
 ## Class Distribution
 
 ```
-marketing: 5 examples (16.7%)
-meals: 5 examples (16.7%)
-office_supplies: 5 examples (16.7%)
-other: 5 examples (16.7%)
-software: 5 examples (16.7%)
-travel: 5 examples (16.7%)
+equipment: 80 examples (20.0%)
+meals: 80 examples (20.0%)
+office_supplies: 80 examples (20.0%)
+other: 81 examples (20.2%)
+travel: 80 examples (20.0%)
 ```
 
 ## Usage
@@ -52,7 +51,7 @@ classifier.add_examples(texts, labels)
 
 ## Training Details
 
-- Training Steps: 15
+- Training Steps: 13
 - Examples per Class: See distribution above
 - Prototype Memory: Active
 - Neural Adaptation: Active

config.json

@@ -33,20 +33,18 @@
   "embedding_dim": 768,
   "id_to_label": {
     "0": "travel",
-    "1": "office_supplies",
-    "2": "meals",
-    "3": "marketing",
-    "4": "software",
-    "5": "other"
+    "1": "other",
+    "2": "office_supplies",
+    "3": "equipment",
+    "4": "meals"
   },
   "label_to_id": {
-    "marketing": 3,
-    "meals": 2,
-    "office_supplies": 1,
-    "other": 5,
-    "software": 4,
+    "equipment": 3,
+    "meals": 4,
+    "office_supplies": 2,
+    "other": 1,
     "travel": 0
   },
   "model_name": "answerdotai/ModernBERT-base",
-  "train_steps": 15
+  "train_steps": 13
 }

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:e52ca396ffe08d3d84a7e9795e341b4bd93c3ccdecef91aa1b2b7dcba8f92d33
-size 3572296
+oid sha256:8bd9d8c832251e2d9d0ef6cae762710c31ac73dec4112761a431a2ae228c862d
+size 3567596