通过embeddings轻松实现搜索功能

前言

随着人工智能和深度学习的进步,智能模型嵌入(embeddings)的引入使得搜索引擎能够通过计算空间维度来发现更为相似的对象。这种方法不仅提升了搜索结果的精确度和个性化程度,还为搜索引擎的进一步优化和创新提供了新的可能性。

embeddings

在计算机科学和机器学习领域,"embeddings"(嵌入)通常指的是将高维空间中的对象(如文本、图像、音频等)映射到低维空间的向量表示。这种向量表示能够捕捉对象的语义和特征,使得它们可以更有效地被计算机处理和理解。具体来说,嵌入是指将对象映射到一个连续的向量空间中,使得相似的对象在这个空间中距离较近。

使用embeddings将文本转换为语义向量

step1:导入所需包

首先,你需要导入必要的Python包。在这个例子中,openai用于与OpenAI的API交互,而.envdotenv用于读取环境变量,这通常用来存储敏感信息如API密钥。

step2:创建OpenAI客户端对象

创建一个连接到OpenAI API的客户端,使得你可以使用该客户端来进行与OpenAI API相关的操作。

step3:调用embeddings接口

通过指定的OpenAI文本嵌入模型,将"如何创建vue组件"这个问题转换为一个向量化的表示,以便后续进行语义相似性比较、文本聚类。

向量化结果:

搜索匹配实现

通过上述操作,一旦数据库中的文档或项目被编码为向量形式并存储于数据库内,当用户发起查询时,其输入的查询文本同样会经过相同的向量化过程,转化为对应的语义向量。随后,系统会对查询向量与数据库中存储的所有向量进行相似度计算,以识别那些在语义上最接近查询意图的条目。

相似度计算常用的方法之一是余弦相似度(cosine similarity),这是一种衡量两个非零向量之间的角度余弦值的度量方式。余弦相似度能够有效地反映两个向量在高维空间中的方向相似性,而不受它们的模长影响,因此非常适合于文本相似性分析。

具体步骤如下:

  1. 向量化:使用预训练的语言模型或专门设计的嵌入算法将文本转化为向量表示。

  2. 存储:将所有文本的向量表示存储在数据库中,优化存储结构以提高检索效率。

  3. 查询向量化:对用户的查询文本进行同样的向量化处理。

  4. 相似度计算:通过计算查询向量与数据库中每个向量的余弦相似度,找到最接近的匹配项。

  5. 排序与输出:根据相似度得分对结果进行排序,并返回得分最高的若干条目作为推荐结果。

demo实例

准备的数据

将这个JSON文件中的数据(每个帖子(post)的标题和分类进行文本嵌入),并将带有嵌入向量的帖子信息保存至另一个文件中,便于后续计算查询向量与数据库中每个向量的余弦相似度。

根据搜索文本 "vue组件开发",找出与其语义最接近的帖子,并按照相似度高低输出相关的帖子标题和分类信息。

结果输出:

最后

最后,我们轻松实现了一个搜索推荐功能,这一机制不仅提升了信息检索的精度和效率,还能够支持基于内容的推荐系统,为用户提供个性化且语义相关的信息或产品建议。

相关推荐
HackTorjan1 小时前
2026年5月29日:全球首个通用人工智能操作系统正式发布,开启人机协同新纪元
人工智能
刘大猫.1 小时前
智造短剧新引擎:火山引擎上线「火山剧创 1.0」,制作效率提升 80%
人工智能·ai·chatgpt·机器人·大模型·火山引擎·短剧新引擎
红尘散仙2 小时前
我把终端小说阅读器接上了 AI Agent:TRNovel 现在能用 skill 生成书源了
人工智能·后端·rust
雅菲奥朗2 小时前
企业级 AI 自动化|OpenClaw 龙虾实战与认证
运维·人工智能·自动化·openclaw
HIT_Weston2 小时前
99、【Agent】【OpenCode】task 工具提示词(Slash command)(一)
人工智能·agent·opencode
25 Hz2 小时前
Mind 爱好者时空表征刊 第24期 | 时间结构学习、空间对时间表征的补偿、事件内部的时间扭曲……
人工智能
心中有国也有家2 小时前
GE图引擎深度解析——CANN的计算图优化与执行引擎
人工智能·pytorch·python·学习·numpy
海兰2 小时前
【文字三国志:第一篇】天命重构,大语言模型(LLM)动态生成文言风格的叙事文本的文字游戏
人工智能·游戏·语言模型
cxr8283 小时前
高分子复合材料 AI 逆向设计合——验证闭环、决策优化与中试放大
人工智能·材料逆向设计合成
litble3 小时前
如何速成LLM以伪装成一个AI研究者(6)——LoRA,Adapter,P-tuning,量化,QLoRA
人工智能·lora·量化·peft·qlora·高效微调