happy transformer
Version 3.0.0
Documentation et actualités : happytransformer.com
Rejoignez notre serveur Discord :
happy transformer facilite l'ajustement et la réalisation d'inférences avec les modèles NLP Transformer.
Découvrez la mise à jour complète 3.0.0, y compris les modifications majeures, ici.
Tâches | Inférence | Entraînement |
---|---|---|
Génération de texte | ✔ | ✔ |
Classement du texte | ✔ | ✔ |
Prédiction de mots | ✔ | ✔ |
Réponse aux questions | ✔ | ✔ |
Texte à texte | ✔ | ✔ |
Prédiction de la phrase suivante | ✔ | |
Classification des jetons | ✔ |
pip install happytransformer
from happytransformer import HappyWordPrediction
#--------------------------------------#
happy_wp = HappyWordPrediction () # default uses distilbert-base-uncased
result = happy_wp . predict_mask ( "I think therefore I [MASK]" )
print ( result ) # [WordPredictionResult(token='am', score=0.10172799974679947)]
print ( result [ 0 ]. token ) # am
Génération de texte avec formation (GPT-Neo)
Classification de textes (formation)
Classification des textes (détection des discours de haine)
Classification de texte (analyse des sentiments)
Prédiction de mots avec entraînement (DistilBERT, RoBERTa)
Meilleurs modèles T5
Correction de grammaire
Affiner un modèle de correction grammaticale