Thrad/thrad-distilbert-conversation-classifier
By Thrad · ↓ 2 · ♥ 2 · Updated 2025-12-14
What's in this repo
This page lists the files in Thrad/thrad-distilbert-conversation-classifier, including model.onnx.
- 1 ONNX models (.onnx files)
- 6 files, including 1 .md, 1 .json, 1 .onnx, 1 .safetensors, 1 .bin
- Last updated 2025-12-14
ONNX files
| File | Size | Download |
|---|---|---|
| model.onnx | — | Download |
How to use
These ONNX files can be loaded with ONNX Runtime. To inspect a model graph, upload the file to the Netron viewer. Download links point to the original files on Hugging Face.
README
thrad-distilbert-conversation-classifier
DistilBERT model for conversation classification with hard labels. For additional information about the data preprocessing, training, or evaluation, refer to the public repository for the project [here].
Model Details
- Base Architecture: DistilBERT
- Task: Multi-class conversation intent classification
Usage
from transformers import AutoTokenizer, AutoModelForSequenceClassification
model = AutoModelForSequenceClassification.from_pretrained("Thrad/thrad-distilbert-conversation-classifier")
tokenizer = AutoTokenizer.from_pretrained("Thrad/thrad-distilbert-conversation-classifier")
# Example inference
text = "Your conversation text here"
inputs = tokenizer(text, return_tensors="pt", truncation=True, max_length=512)
outputs = model(**inputs)
predictions = outputs.logits.softmax(dim=-1)
Model info
Created: 2025-11-09
Updated: 2025-12-14
View on Hugging FaceFiles (6)
Related models
More from Thrad
No other models by this author.