prodyge94/td5-query-classifier

DistilBERT (distilbert-base-uncased) fine-tuné en SequenceClassification pour router une requête utilisateur vers question_rag (question sur un cours -> RAG) ou send_message (envoyer un message).

Entraîné sur 78 requêtes (20 en test, accuracy 100 %), déclinées avec / sans ponctuation finale pour que le modèle ne s'appuie pas sur le "?". Seules la dernière couche du transformer et la tête de classification ont été entraînées, le reste du modèle est gelé.

Projet : NLP TD5 - Virtual assistant.