训练与查询
训练与查询
import faiss, numpy as np
d = 384; nlist = 1024
quantizer = faiss.IndexFlatL2(d)
index = faiss.IndexIVFFlat(quantizer, d, nlist, faiss.METRIC_L2)
index.train(np.random.rand(200000, d).astype("float32"))
index.add(np.random.rand(1000000, d).astype("float32"))
index.nprobe = 32
D, I = index.search(np.random.rand(1, d).astype("float32"), 10)训练样本量与 nlist 同阶或更多时聚类更稳定；过少会让部分聚类形同虚设。