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

商汤科技词嵌入模型登顶中文embedding排行榜

2023-09-14 21:54 作者:AI日日新  | 我要投稿

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

piccolo-large-zh


首先需要安装依赖包:

接着即可加载模型:

然后我提供两个句子计算它们的相似度:

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

试用完模型推理后,我们还可以微调模型,得到符合我们使用场景的新模型,那么我们首先准备数据:

得到需要的数据后即可开始模型微调:

在完成微调后,我们还需要测试微调模型的效果,我提供训练的数据用于测试:

最终我们得到的相似度较高,所以效果还可以。


微调步骤的notebook也可在 kaggle 搜索:

finetuned piccolo-base-zh


商汤科技词嵌入模型登顶中文embedding排行榜的评论 (共 条)

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