IndexFlatIP 找相似(内积)
IndexFlatIP 找相似(内积)
import faiss, numpy as np
xb = np.random.rand(100000, 384).astype("float32")
faiss.normalize_L2(xb) # 归一化后内积等价余弦
index = faiss.IndexFlatIP(384)
index.add(xb)
q = np.random.rand(1, 384).astype("float32")
faiss.normalize_L2(q)
D, I = index.search(q, 5) # D=相似度, I=向量序号
print(I[0], D[0])Flat 索引无需训练、可随时添加;替换为 IVF/HNSW 前先以它为召回基线。