上传者: 42132359
|
上传时间: 2021-10-02 19:01:24
|
文件大小: 720KB
|
文件类型: -
EasyBert
基于Pytorch的Bert应用,包括命名实体识别、情感分析、文本分类以及文本相似度等(后续更新其他方向相关模块),并有相关数据与深度训练优化方式api。各个子项目大都为开源工作,本项目仅做相应处理以及提供一个已训练的预测接口,方便需求者进行快速的使用。本项目仅作为学习与研究使用,若存在侵权行为,请原作者务必联系我进行协商处理。
使用示例
使用前需下载相应的已训练模型,并导入相应位置
模型下载地址:
在此目录下各以需求名命名的文件中提供相应的使用演示,本阶段所训练的模型效果可以满足相应任务的基本需求。
现阶段通过各任务接口的时间相对慢,大都是在模型加载阶段。若想提升相应的速度,请使用者在接受相应精度损失的前提下更换AlBert进行相应任务的重新预训练。
依赖项
环境依赖
python >= 3.7 Pytorch >= 1.14 transformers >= 2.8.0