happy transformer
Version 3.0.0
Documentação e notícias: happytransformer.com
Junte-se ao nosso servidor Discord:
happy transformer facilita o ajuste fino e a realização de inferências com modelos do NLP Transformer.
Leia sobre a atualização 3.0.0 completa, incluindo alterações importantes aqui.
Tarefas | Inferência | Treinamento |
---|---|---|
Geração de texto | ✔ | ✔ |
Classificação de texto | ✔ | ✔ |
Previsão de palavras | ✔ | ✔ |
Resposta a perguntas | ✔ | ✔ |
Texto para Texto | ✔ | ✔ |
Previsão da próxima frase | ✔ | |
Classificação 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
Geração de texto com treinamento (GPT-Neo)
Classificação de texto (treinamento)
Classificação de texto (detecção de discurso de ódio)
Classificação de texto (análise de sentimento)
Previsão de palavras com treinamento (DistilBERT, RoBERTa)
Principais modelos T5
Correção gramatical
Ajustar um modelo de correção gramatical