构建高效搜索系统 - Faiss向量数据库的快速入门

目录

快速入门

创建第一个Faiss索引

加载数据到索引中

执行基本查询

评估索引性能


快速入门

创建第一个Faiss索引

先需要导入必要的库,并定义一个索引对象。使用最基础的Flat索引作为例子。

bash 复制代码
import numpy as np
import faiss

# 设置向量的维度
d = 128

# 创建一个Flat索引,使用L2(欧几里得)距离
index = faiss.IndexFlatL2(d)

# 打印索引信息
print("Index created:", index)

先导入numpyfaiss库。指定了向量的维度为128,并创建一个基于L2距离的Flat索引对象。IndexFlatL2是最简单的索引类型,会在内存中存储所有的向量,并计算所有向量间的距离来找出最近邻。

加载数据到索引中

需要生成一些随机向量数据,并将其添加到刚刚创建的索引中。

bash 复制代码
# 生成10000个随机向量
nb = 10000
np.random.seed(1234)  # 设置随机种子以确保每次运行都得到相同结果
xb = np.random.random((nb, d)).astype('float32')

# 将向量数据归一化到单位长度
xb /= np.linalg.norm(xb, axis=1, keepdims=True)

# 添加向量到索引
index.add(xb)
print("Vectors added to index.")

在这个步骤中,先设定了要添加的向量数量为10000个,并生成这些向量。为使距离度量更加有效,将向量进行了归一化处理。最后调用了add方法将这些向量添加到了索引中。

执行基本查询

可以尝试使用一些查询向量来测试索引是否正常工作。

bash 复制代码
# 生成10个查询向量
nq = 10
xq = np.random.random((nq, d)).astype('float32')
xq /= np.linalg.norm(xq, axis=1, keepdims=True)

# 执行搜索,返回每个查询向量的k个最近邻
k = 4
D, I = index.search(xq, k)

# 输出结果
print("Distances:")
print(D)
print("Indices:")
print(I)

这里生成了10个查询向量,并设置了返回最近邻的数量为4。index.search函数执行了实际的搜索操作,并返回了两组结果:D表示查询向量到最近邻的距离,I表示这些最近邻的索引号。

评估索引性能

为了评估索引的性能,可以测量查询所需的时间,并检查返回结果的正确性。

bash 复制代码
import time

# 测量搜索耗时
start_time = time.time()
D, I = index.search(xq, k)
end_time = time.time()

# 计算查询时间
search_time = end_time - start_time
print(f"Search took {search_time:.4f} seconds.")

# 检查结果是否合理
print("Checking results...")
assert D.shape == (nq, k)
assert I.shape == (nq, k)
print("Results are valid.")

在这段代码中使用Python的time模块来记录搜索操作的起始和结束时间,从而计算出总的查询耗时。还通过断言检查了返回结果的形状是否符合预期,以此验证结果的有效性。

相关推荐
多吃轻食3 天前
向量数据库FAISS之二:基础进阶版
数据库·faiss
多吃轻食3 天前
向量数据库FAISS之六:如何让FAISS更快
数据库·faiss
多吃轻食3 天前
向量数据库FAISS之五:原理(LSH、PQ、HNSW、IVF)
数据库·人工智能·深度学习·语言模型·自然语言处理·faiss
多吃轻食3 天前
向量数据库FAISS之四:向量检索和 FAISS
数据库·faiss
weixin_487058413 天前
使用Faiss构建音频特征索引并计算余弦相似度
音视频·faiss
多吃轻食3 天前
向量数据库FAISS之一:官方简单教程
数据库·python·深度学习·语言模型·自然语言处理·faiss
weixin_487058414 天前
faiss 提供了多种索引类型
人工智能·机器学习·faiss
CrazyCosin4 天前
resnet50,clip,Faiss+Flask简易图文搜索服务
python·flask·faiss·clip·resnet50
CrazyCosin7 天前
Clip结合Faiss+Flask简易版文搜图服务
python·flask·faiss
科研小达人17 天前
Langchain调用模型使用FAISS
python·chatgpt·langchain·faiss