Happy Transformer
文档和新闻: happytransformer.com
Happy Transformer 使得微调和执行NLP Transformer模型的推理变得简单。
3.0.0
- 用于训练的DeepSpeed
- 用于训练和推理的苹果MPS
- 使用WandB来跟踪训练过程
- 提供的训练数据自动分割成训练和评估部分
- 直接将模型推送到Hugging Face的模型中心
阅读有关3.0.0更新的全部内容,包括重要改变 这里。
任务
任务 | 推理 | 训练 |
---|---|---|
文本生成 | ✔ | ✔ |
文本分类 | ✔ | ✔ |
词预测 | ✔ | ✔ |
问答 | ✔ | ✔ |
文本到文本 | ✔ | ✔ |
下一句预测 | ✔ | |
标记分类 | ✔ |
快速开始
pip install happytransformer
from happytransformer import HappyWordPrediction
#--------------------------------------#
happy_wp = HappyWordPrediction() # 默认使用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
维护人员
- Eric Fillion 主要维护者
- Ted Brownlow 维护者