当前位置：首页 > news >正文

【拥抱AI】向量数据库有哪些常见的检索算法？

news 2025/6/30 0:43:14

在信息检索领域，有许多常见的算法用于帮助用户从大量数据中找到相关的信息。以下是一些常见的检索算法：

布尔模型示例（文本操作）

在文本操作中，布尔模型可以通过编写一个简单的脚本来实现。例如，你可以创建一个包含多个文档的文本文件，然后编写一个脚本来搜索包含特定关键词的文档。

# 布尔模型示例
# 假设我们有以下文档
documents = ["苹果手机 苹果手机价格","苹果手机介绍","华为手机介绍","小米手机介绍"
]
# 用户查询
query = "苹果手机"
# 布尔模型搜索
results = []
for document in documents:if query in document:results.append(document)
print(results)

向量空间模型示例（文本操作）

在向量空间模型中，你可以使用Python的nltk库来计算文档和查询的向量表示，并计算它们的相似度。

from nltk.corpus import stopwords
from nltk.tokenize import word_tokenize
from nltk.probability import FreqDist
from nltk.util import ngrams
from sklearn.feature_extraction.text import TfidfVectorizer
# 假设我们有以下文档
documents = ["苹果手机 苹果手机价格","苹果手机介绍","华为手机介绍","小米手机介绍"
]
# 用户查询
query = "苹果手机"
# 文本预处理
def preprocess_text(text):stop_words = set(stopwords.words('english'))words = word_tokenize(text.lower())return [word for word in words if word not in stop_words]
# 计算TF-IDF向量
vectorizer = TfidfVectorizer(preprocessor=preprocess_text)
tfidf_matrix = vectorizer.fit_transform(documents + [query])
# 计算相似度
cosine_similarities = tfidf_matrix * tfidf_matrix.T
# 返回相似度最高的文档
results = cosine_similarities.toarray()
print(results)

倒排索引示例（文本操作）

在倒排索引中，你可以使用Python的collections模块来创建一个简单的倒排索引。

from collections import defaultdict
# 假设我们有以下文档
documents = ["苹果手机 苹果手机价格","苹果手机介绍","华为手机介绍","小米手机介绍"
]
# 创建倒排索引
inverted_index = defaultdict(list)
for document in documents:words = document.split()for word in words:inverted_index[word].append(document)
# 打印倒排索引
print(inverted_index)

基于词频-逆文档频率（TF-IDF）的检索：

*** 基于词频-逆文档频率（TF-IDF）的检索是一种常用的文本信息检索技术，它通过计算词的TF-IDF值来衡量词的重要性，并据此对文档进行排序。TF-IDF算法的基本思想是，如果一个词在某个文档中出现的频率较高（词频，Term Frequency，TF），并且在整个文档集合中出现的频率较低（逆文档频率，Inverse Document Frequency，IDF），那么这个词对于这个文档的贡献较大，应该被赋予较高的权重。

计算步骤

词频（TF）计算：
- 对于文档中的每个词，计算它在文档中出现的次数。
逆文档频率（IDF）计算：
- 计算整个文档集合中包含这个词的文档数。
- 计算文档总数。
- 计算IDF值，公式为：[ IDF = \log_2 \frac{文档总数}{包含这个词的文档数} ]
TF-IDF计算：
- 对于文档中的每个词，计算它的TF-IDF值，公式为：[ TF-IDF = TF \times IDF ]
文档向量表示：
- 将文档中每个词的TF-IDF值作为文档向量的一个维度。
相似度计算：
- 使用余弦相似度等方法计算查询向量与文档向量之间的相似度。
- 返回相似度最高的文档。

示例说明

假设我们有一个包含以下文档的文档集合：

文档1: 苹果手机 苹果手机价格
文档2: 苹果手机介绍
文档3: 华为手机介绍
文档4: 小米手机介绍

用户查询为“苹果手机”。

词频（TF）计算：
- 在文档1中，“苹果手机”出现了2次，TF值为2。
- 在文档2中，“苹果手机”出现了1次，TF值为1。
- 在文档3和文档4中，“苹果手机”没有出现，TF值为0。
逆文档频率（IDF）计算：
- 在文档集合中，包含“苹果手机”的文档数为2（文档1和文档2）。
- 文档总数为4。
- IDF值为 [ \log_2 \frac{4}{2} = \log_2 2 = 1 ]
TF-IDF计算：
- 在文档1中，“苹果手机”的TF-IDF值为 [ 2 \times 1 = 2 ]
- 在文档2中，“苹果手机”的TF-IDF值为 [ 1 \times 1 = 1 ]
- 在文档3和文档4中，“苹果手机”的TF-IDF值为0。
文档向量表示：
- 文档1的向量为 [2, 0, 0, 0]。
- 文档2的向量为 [1, 0, 0, 0]。
- 文档3和文档4的向量为 [0, 0, 0, 0]。
相似度计算：
- 假设查询向量为 [1, 0, 0, 0]。
- 计算文档1和查询之间的余弦相似度为 [ \frac{2}{2} = 1 ]
- 计算文档2和查询之间的余弦相似度为 [ \frac{1}{2} = 0.5 ]
- 文档3和文档4与查询之间的余弦相似度为0。
返回结果：
- 基于余弦相似度，文档1与查询最相关，其次是文档2。