happy transformer
Version 3.0.0
Documentación y novedades: happytransformer.com
Únase a nuestro servidor de Discord:
happy transformer facilita el ajuste y la realización de inferencias con los modelos NLP Transformer.
Lea acerca de la actualización 3.0.0 completa, incluidos los cambios importantes, aquí.
Tareas | Inferencia | Capacitación |
---|---|---|
Generación de texto | ✔ | ✔ |
Clasificación de texto | ✔ | ✔ |
Predicción de palabras | ✔ | ✔ |
Respuesta a preguntas | ✔ | ✔ |
Texto a texto | ✔ | ✔ |
Predicción de la siguiente oración | ✔ | |
Clasificación de tokens | ✔ |
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
Generación de texto con entrenamiento (GPT-Neo)
Clasificación de textos (formación)
Clasificación de texto (detección de discurso de odio)
Clasificación de texto (análisis de sentimiento)
Predicción de palabras con entrenamiento (DistilBERT, RoBERTa)
Mejores modelos T5
Corrección gramatical
Ajustar un modelo de corrección gramatical