构建高效搜索系统 - 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模块来记录搜索操作的起始和结束时间,从而计算出总的查询耗时。还通过断言检查了返回结果的形状是否符合预期,以此验证结果的有效性。

相关推荐
科研小达人2 天前
Langchain调用模型使用FAISS
python·chatgpt·langchain·faiss
陌北v110 天前
Milvus 与 Faiss:选择合适的向量数据库
milvus·faiss·向量数据库
江无羡2 个月前
FastText 和 Faiss 的初探了解
faiss
何遇mirror2 个月前
【Faiss】构建高效搜索系统 - Faiss向量数据库的搭建
faiss
程序者王大川2 个月前
【大数据】深入解析向量数据库Faiss:搭建与使用指南
数据库·学习·ai编程·kylin·faiss
微学AI2 个月前
人工智能任务6-基于FAISS数据库的应用:向量数据库的搭建与中文文本相似度搜索
数据库·人工智能·faiss
大白菜程序猿2 个月前
向量数据库Faiss的详细介绍和搭建使用教程
faiss
王卫东2 个月前
ES 模糊查询 wildcard 的替代方案探索
elasticsearch·faiss·es·ngram·wildcard·模糊搜索
cesske2 个月前
springboot springmvc spring区别
faiss