Text Classification
PEFT
Safetensors
English
intent-classification
ecommerce
lora
qlora
function-calling
Instructions to use while-ai/ecommerce-1b with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- PEFT
How to use while-ai/ecommerce-1b with PEFT:
from peft import PeftModel from transformers import AutoModelForCausalLM base_model = AutoModelForCausalLM.from_pretrained("unsloth/gemma-3-1b-it-bnb-4bit") model = PeftModel.from_pretrained(base_model, "while-ai/ecommerce-1b") - Notebooks
- Google Colab
- Kaggle
Card: One name (While, whileai SDK), current links
#1
by whileai - opened
README.md
CHANGED
|
@@ -27,7 +27,7 @@ On a hard, held-out benchmark it reaches **75.3% macro-averaged intent-type accu
|
|
| 27 |
|
| 28 |
### Model Description
|
| 29 |
|
| 30 |
-
- **Developed by:** While
|
| 31 |
- **Model type:** E-commerce payment-intent classifier; structured JSON output over seven intent types
|
| 32 |
- **Language:** English
|
| 33 |
- **License:** Gemma (inherited from the base model)
|
|
@@ -85,7 +85,7 @@ The model returns one JSON object per message:
|
|
| 85 |
|
| 86 |
### Training Data
|
| 87 |
|
| 88 |
-
The differentiator is the data. While
|
| 89 |
|
| 90 |
This release was fine-tuned on **17,144 conversations**. The held-out evaluation set contains **1,977 conversations** and has no conversation or identifier overlap with the training split.
|
| 91 |
|
|
@@ -107,7 +107,7 @@ Held-out eval of 1,977 conversations, zero train/eval leakage, macro-averaged (e
|
|
| 107 |
|
| 108 |
### Results
|
| 109 |
|
| 110 |
-

|
| 111 |
|
| 112 |
- Fine-tuning takes intent-type accuracy from **14.4% to 75.3%** on the public six-action rubric, macro-averaged across the 1,977-row held-out evaluation.
|
| 113 |
- Intent detection reaches **80.7%**, and structured order-detail extraction reaches **66.3%**, both macro-averaged on the same public rubric.
|
|
|
|
| 126 |
|
| 127 |
## Model Card Contact
|
| 128 |
|
| 129 |
+
While, https://huggingface.co/while-ai
|