happy transformer
Version 3.0.0
文档和新闻:happytransformer.com
加入我们的 Discord 服务器:
happy transformer可以轻松地使用 NLP Transformer 模型进行微调和执行推理。
在此处阅读完整的 3.0.0 更新,包括重大更改。
任务 | 推理 | 训练 |
---|---|---|
文本生成 | ✔ | ✔ |
文本分类 | ✔ | ✔ |
单词预测 | ✔ | ✔ |
问答 | ✔ | ✔ |
文本到文本 | ✔ | ✔ |
下一句话预测 | ✔ | |
代币分类 | ✔ |
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
通过训练生成文本(GPT-Neo)
文本分类(训练)
文本分类(仇恨言论检测)
文本分类(情感分析)
通过训练进行单词预测(DistilBERT、RoBERTa)
顶级 T5 型号
语法纠正
微调语法校正模型