You can deploy this model and start earning money today!

Yacine75/
question-classifier

Views

No views yet

🤗 DistilBERT - Classification de Questions

Ce modèle est un DistilBERT fine-tuné pour classifier les requêtes en :

question_rag → ex: "How can I enroll in the Python course?"
send_message → ex: "Send a message to my friend."

📌 Comment l'utiliser ?

python

from transformers import pipeline

classifier = pipeline("text-classification", model="Yacine75/question-classifier")
result = classifier("How can I enroll in the Python course?")
print(result)

📊 Performances

Accuracy : 97.8%
Validation Loss : 0.0246

📌 Détails de l'entraînement

Modèle de base : distilbert-base-uncased
Tâche : Classification binaire (question_rag / send_message)
Dataset : 98 exemples annotés
Époques : 3
Optimiseur : AdamW
Batch size : 8

🚀 Liens utiles

🔗 Voir le modèle sur Hugging Face