File size: 1,284 Bytes
6f8e194 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 | ---
language: en
tags:
- text-classification
- conversational-ai
- thought-completion
widget:
- text: "Great. Would you like anything else? [SEP] Picking up on app order."
example_title: "Complete thought"
- text: "Your total is $15.99 [SEP] Actually, let me add"
example_title: "Incomplete thought"
---
# Thought Completion Classifier
This model classifies whether a conversation between an AI assistant and a customer represents a complete or incomplete thought.
## Model Description
- **Model Type:** DistilBERT fine-tuned for sequence classification
- **Task:** Binary classification (Complete/Incomplete thought)
- **Training Data:** 900+ labeled conversation examples
- **Input Format:** "AI_Utterance [SEP] CX_Utterance"
## Usage
```python
from transformers import pipeline
# Load the model
classifier = pipeline("text-classification", model="samurai9776/thought-classifier")
# Example usage
result = classifier("What else can I get you? [SEP] That's all for now")
print(result)
# Output: [{'label': 'COMPLETE', 'score': 0.95}]
```
## Training Details
- Base Model: distilbert-base-uncased
- Training epochs: 3
- Batch size: 16
- Learning rate: 2e-5
## Limitations
- Trained on English conversations only
- Optimized for customer service/order-taking scenarios
|