目录
快速入门
创建第一个Faiss索引
先需要导入必要的库,并定义一个索引对象。使用最基础的Flat索引作为例子。
bash
import numpy as np
import faiss
# 设置向量的维度
d = 128
# 创建一个Flat索引,使用L2(欧几里得)距离
index = faiss.IndexFlatL2(d)
# 打印索引信息
print("Index created:", index)
先导入numpy
和faiss
库。指定了向量的维度为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
模块来记录搜索操作的起始和结束时间,从而计算出总的查询耗时。还通过断言检查了返回结果的形状是否符合预期,以此验证结果的有效性。