TD5 DistilBERT classifier

Classifie une requête anglaise en question_rag ou send_message.

DistilBERT entraîné sur les 98 questions du TD5 ESGI, avec le dernier bloc et la tête de classification dégelés, pendant 5 époques. Validation croisée stratifiée à 5 plis : accuracy de 100 % sur chaque pli. Le modèle final est ensuite entraîné sur les 98 questions. Ce score sur un petit dataset ne garantit pas la généralisation : les demandes indirectes peuvent être mal classées.

Downloads last month
13
Safetensors
Model size
67M params
Tensor type
F32
·
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support

Model tree for Nisseya/td5-distilbert-classifier

Finetuned
(12691)
this model
Finetunes
1 model