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

某大神在商汤的词嵌入模型基础上训练出了更强模型

2023-09-19 21:20 作者:AI日日新  | 我要投稿

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

stella-base-zh


首先安装依赖:

接着加载该模型:

那么我们就可以计算两段文本的相似度了:

有时候直接使用该模型无法达到你的预期水平,可以考虑微调该模型。首先我们需要准备数据:

接着使用上面准备的数据微调模型:

接着仍然是计算两段文本的相似度,最终的相似度有所提高:


某大神在商汤的词嵌入模型基础上训练出了更强模型的评论 (共 条)

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