ToxiCheck / models /zero_shot.py

Commit History

feat(models): retourne tous les labels et scores dans ZeroShotModel.predict
0005ea7

Tbruand commited on

feat(models): modifie ZeroShotModel pour retourner le label et le score de confiance
7301ed6

Tbruand commited on

feat(models): ajoute un modèle de classification zero-shot basé sur bart-large-mnli
9decc9d

Tbruand commited on

chore: initial commit - project structure
2423876

Tbruand commited on