Text Classification
PEFT
Safetensors
English
intent-classification
ecommerce
lora
qlora
function-calling
Instructions to use while-ai/ecommerce-0.5b with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- PEFT
How to use while-ai/ecommerce-0.5b with PEFT:
from peft import PeftModel from transformers import AutoModelForCausalLM base_model = AutoModelForCausalLM.from_pretrained("unsloth/qwen2.5-0.5b-instruct-bnb-4bit") model = PeftModel.from_pretrained(base_model, "while-ai/ecommerce-0.5b") - Notebooks
- Google Colab
- Kaggle
Card: One name (While, whileai SDK), current links
#1
by whileai - opened
README.md
CHANGED
|
@@ -17,7 +17,7 @@ language:
|
|
| 17 |
|
| 18 |
# Model Card for ecommerce-0.5b
|
| 19 |
|
| 20 |
-
**E-commerce intent detection in a 0.5B model, the smallest member of the While
|
| 21 |
|
| 22 |
On a hard, held-out benchmark it reaches **70.3 macro-averaged intent-type accuracy, roughly 91% of our 1B**, from a base model that scores 10.7.
|
| 23 |
|
|
@@ -25,7 +25,7 @@ On a hard, held-out benchmark it reaches **70.3 macro-averaged intent-type accur
|
|
| 25 |
|
| 26 |
### Model Description
|
| 27 |
|
| 28 |
-
- **Developed by:** While
|
| 29 |
- **Model type:** E-commerce payment-intent classifier; structured JSON output over seven intent types
|
| 30 |
- **Language:** English
|
| 31 |
- **License:** Apache-2.0 (inherited from the base model)
|
|
@@ -70,7 +70,7 @@ Use greedy decoding (`temperature=0`). The system prompt and per-intent detail s
|
|
| 70 |
|
| 71 |
### Training Data
|
| 72 |
|
| 73 |
-
The differentiator is the data. While
|
| 74 |
|
| 75 |
### Training Procedure
|
| 76 |
|
|
@@ -102,4 +102,4 @@ A LoRA adapter over `Qwen2.5-0.5B-Instruct`, trained response-only to emit one s
|
|
| 102 |
|
| 103 |
## Model Card Contact
|
| 104 |
|
| 105 |
-
While
|
|
|
|
| 17 |
|
| 18 |
# Model Card for ecommerce-0.5b
|
| 19 |
|
| 20 |
+
**E-commerce intent detection in a 0.5B model, the smallest member of the While family.** Given a live customer conversation, `ecommerce-0.5b` returns a single JSON object with the customer's payment intent and its structured details, at the lowest cost and latency in the family.
|
| 21 |
|
| 22 |
On a hard, held-out benchmark it reaches **70.3 macro-averaged intent-type accuracy, roughly 91% of our 1B**, from a base model that scores 10.7.
|
| 23 |
|
|
|
|
| 25 |
|
| 26 |
### Model Description
|
| 27 |
|
| 28 |
+
- **Developed by:** While
|
| 29 |
- **Model type:** E-commerce payment-intent classifier; structured JSON output over seven intent types
|
| 30 |
- **Language:** English
|
| 31 |
- **License:** Apache-2.0 (inherited from the base model)
|
|
|
|
| 70 |
|
| 71 |
### Training Data
|
| 72 |
|
| 73 |
+
The differentiator is the data. While builds e-commerce intent training data as **randomized conversational simulations**: role-played customers with independently sampled personas, situations, and behaviors including adversarial actors, simulated turn by turn between two independently drawn models. Generation is label-blind, labels are assigned in a separate pass, and every candidate passes a structural data gate with zero train/eval leakage. This 0.5B was trained on the same class-rebalanced split as the 1B so it routes across all intent types rather than defaulting to the majority.
|
| 74 |
|
| 75 |
### Training Procedure
|
| 76 |
|
|
|
|
| 102 |
|
| 103 |
## Model Card Contact
|
| 104 |
|
| 105 |
+
While, https://huggingface.co/while-ai
|