基于Pytorch的Bert应用,包括命名实体识别、情感分析、文本分类以及文本相似度等(后续更新其他方向相关模块),并有相关数据与深度训练优化方式api。各个子项目大都为开源工作,本项目仅做相应处理以及提供一个已训练的预测接口,方便需求者进行快速的使用。
使用示例
项目 代码,数据集 获取方式:
关注微信公众号 datayx 然后回复 bert 即可获取。
AI项目体验地址 https://loveai.tech
python >= 3.7 Pytorch >= 1.14 transformers >= 2.8.0
注:作者实验环境,其他环境未测试
原始训练数据:该部分原数据因项目原因不提供,如需重新训练可更换其他开源数据集.
相关源代码已有优秀开源,本项目借鉴训练时部分加入相关训练优化trcik,部分保持原始代码复现格式,若需相关训练优化功能,相关代码及使用方式已给出,按需使用。
机器学习算法AI大数据技术
搜索公众号添加: datanlp