Push model using huggingface_hub.

README.md

@@ -17,15 +17,15 @@ You can install it with `pip install adaptive-classifier`.
 
 - Base Model: distilbert-base-uncased
 - Number of Classes: 3
-- Total Examples: 150
+- Total Examples: 600
 - Embedding Dimension: 768
 
 ## Class Distribution
 
 ```
-negative: 50 examples (33.3%)
-neutral: 50 examples (33.3%)
-positive: 50 examples (33.3%)
+negative: 200 examples (33.3%)
+neutral: 200 examples (33.3%)
+positive: 200 examples (33.3%)
 ```
 
 ## Usage
@@ -49,7 +49,7 @@ classifier.add_examples(texts, labels)
 
 ## Training Details
 
-- Training Steps: 28
+- Training Steps: 14
 - Examples per Class: See distribution above
 - Prototype Memory: Active
 - Neural Adaptation: Active
@@ -57,9 +57,9 @@ classifier.add_examples(texts, labels)
 ## Limitations
 
 This model:
-- Requires at least 5 examples per class
-- Has a maximum of 50 examples per class
-- Updates prototypes every 25 examples
+- Requires at least 10 examples per class
+- Has a maximum of 200 examples per class
+- Updates prototypes every 50 examples
 
 ## Citation
 

config.json

@@ -6,14 +6,14 @@
     "epochs": 10,
     "ewc_lambda": 100.0,
     "gradient_checkpointing": false,
-    "learning_rate": 0.001,
-    "max_examples_per_class": 50,
+    "learning_rate": 0.0005,
+    "max_examples_per_class": 200,
     "max_length": 128,
     "min_confidence": 0.1,
-    "min_examples_per_class": 5,
+    "min_examples_per_class": 10,
     "neural_weight": 0.3,
-    "num_representative_examples": 10,
-    "prototype_update_frequency": 25,
+    "num_representative_examples": 20,
+    "prototype_update_frequency": 50,
     "prototype_weight": 0.7,
     "quantization": null,
     "similarity_threshold": 0.6,
@@ -21,15 +21,15 @@
   },
   "embedding_dim": 768,
   "id_to_label": {
-    "0": "neutral",
-    "1": "positive",
-    "2": "negative"
+    "0": "positive",
+    "1": "negative",
+    "2": "neutral"
   },
   "label_to_id": {
-    "negative": 2,
-    "neutral": 0,
-    "positive": 1
+    "negative": 1,
+    "neutral": 2,
+    "positive": 0
   },
   "model_name": "distilbert-base-uncased",
-  "train_steps": 28
+  "train_steps": 14
 }

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:19381b0916d2223edf789cc7d472a6fc52db77a718bac7300ef938343ecb23fc
+oid sha256:c04dd75a39ee27a591fba46a79584018cb5d085cde42b558cb020059b896f392
 size 3558204