
EasyBert:利用PyTorch实现的BERT应用,涵盖命名实体识别、情感分析、文本分类及文本相似度等功能
5星
- 浏览量: 0
- 大小:None
- 文件类型:None
简介:
EasyBert是一款基于PyTorch框架开发的便捷BERT工具包,集成了命名实体识别、情感分析、文本分类和计算文本相似度等实用功能。
EasyBert基于Pytorch的Bert应用,包括命名实体识别、情感分析、文本分类以及文本相似度等功能(后续会更新其他相关模块),并提供数据与深度训练优化方式的相关API。各个子项目大多为开源工作,本项目仅做相应处理,并提供一个已训练好的预测接口,方便用户快速使用。本项目仅供学习和研究使用,若存在侵权行为,请原作者联系我进行协商处理。
在使用前需下载相应的预训练模型并导入到指定位置。各任务的使用示例可以在需求命名的文件中找到,在现阶段所训练的模型可以满足相应任务的基本需求。目前通过各个任务接口的时间相对较慢,主要是因为模型加载阶段耗时较长。如果想提高速度,请用户在可接受精度损失的前提下更换AlBert进行重新预训练。
依赖项:
- Python >= 3.7
- Pytorch >= 1.14
- transformers >= 2.8.0
全部评论 (0)
还没有任何评论哟~


