基于Sentence Transformers和faiss文本检索与匹配算法

1、整体框架介绍

在这里插入图片描述
本次项目采用Sentence Transformers特征提取模型和深度学习向量库faiss的方法去实现文本检索与匹配。其中输入的文本数据流,包含离线和实时的数据,通过特征提取模型将提取的特征存入npy文件和faiss向量库中,利用faiss中的余弦相似度去计算是否相似。

1.1、SentenceTransformers 库介绍

SentenceTransformers 是一个可以用于句子、文本和图像嵌入的Python库。 可以为 100 多种语言计算文本的嵌入并且可以轻松地将它们用于语义文本相似性、语义搜索和同义词挖掘等常见任务。
该框架基于 PyTorch 和 Transformers,并提供了大量针对各种任务的预训练模型。 还可以很容易根据自己的模型进行微调。
阅读论文 Sentence-BERT: Sentence Embeddings using Siamese BERT-Networks,深入了解模型的训练方式。 在本文中,我们将看到该库的一些可能用例的代码示例。
在这里插入图片描述
具体细节请参考:https://blog.csdn.net/deephub/article/details/122516399

1.2、faiss介绍

  • 21
    点赞
  • 19
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值