Add document-type

README.md

@@ -16,20 +16,18 @@ You can install it with `pip install adaptive-classifier`.
 ## Model Details
 
 - Base Model: answerdotai/ModernBERT-base
-- Number of Classes: 7
-- Total Examples: 544
+- Number of Classes: 5
+- Total Examples: 402
 - Embedding Dimension: 768
 
 ## Class Distribution
 
 ```
-contract: 80 examples (14.7%)
-invoice: 80 examples (14.7%)
-memo: 80 examples (14.7%)
-other: 69 examples (12.7%)
-proposal: 80 examples (14.7%)
-receipt: 80 examples (14.7%)
-report: 75 examples (13.8%)
+contract: 81 examples (20.1%)
+email: 80 examples (19.9%)
+invoice: 80 examples (19.9%)
+memo: 80 examples (19.9%)
+report: 81 examples (20.1%)
 ```
 
 ## Usage
@@ -53,7 +51,7 @@ classifier.add_examples(texts, labels)
 
 ## Training Details
 
-- Training Steps: 17
+- Training Steps: 13
 - Examples per Class: See distribution above
 - Prototype Memory: Active
 - Neural Adaptation: Active

config.json

@@ -32,23 +32,19 @@
   },
   "embedding_dim": 768,
   "id_to_label": {
-    "0": "other",
-    "1": "report",
-    "2": "memo",
-    "3": "proposal",
-    "4": "invoice",
-    "5": "contract",
-    "6": "receipt"
+    "0": "report",
+    "1": "contract",
+    "2": "invoice",
+    "3": "email",
+    "4": "memo"
   },
   "label_to_id": {
-    "contract": 5,
-    "invoice": 4,
-    "memo": 2,
-    "other": 0,
-    "proposal": 3,
-    "receipt": 6,
-    "report": 1
+    "contract": 1,
+    "email": 3,
+    "invoice": 2,
+    "memo": 4,
+    "report": 0
   },
   "model_name": "answerdotai/ModernBERT-base",
-  "train_steps": 17
+  "train_steps": 13
 }

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:71cd1d87b7d7b7996c3b9ba86adba2c8cd40b8c02e5e46c3bd834e622f259050
-size 3576980
+oid sha256:347ac02db35948a0816c12eb74e9f984d5b74130f287346a1a306e636cec0adb
+size 3567588