欢迎光临散文网 会员登陆 & 注册

智源发布新版bge词嵌入模型

2023-09-15 21:04 作者:AI日日新  | 我要投稿

可在线运行的notebook:在kaggle网站搜索

BAAI/bge-small-zh-v1.5


首先安装所需依赖:

接着对模型进行推理:

最终推理得到的文本相似度较为合理

模型推理有时候可能不能符合我们的业务场景,那么可以对模型进行微调。我们首先准确用于微调的数据:

接着即可对模型进行微调:

完成模型微调后,我们加载新模型进行测试:

最终我们得到的文本相似度为0.7,较为合理

智源发布新版bge词嵌入模型的评论 (共 条)

分享到微博请遵守国家法律