Yacine75/question-classifier

🤗 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 ?

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