faiss使用add_with_ids

0.环境

ubuntu16.04
faiss-gpu=1.6.3

1.IndexFlatIP使用add_with_ids

dimension = 30
index = faiss.IndexFlatIP(dimension)
index = faiss.IndexIDMap(index) #because IndexFlatIP do not support add_with_ids
index.add_with_ids(g, g_pids) # works, the vectors are stored in the underlying index
k = 1
Distance, Index = index.search(g[:100], k)

2.IndexIVFFlat使用add_with_ids

nlist = 10 #聚类中心
dimension = 30
quantizer  = faiss.IndexFlatIP(dimension)
index = faiss.IndexIVFFlat(quantizer, dimension, nlist, faiss.METRIC_INNER_PRODUCT)
index.train(g)
index.add_with_ids(g, g_pids)
k = 1
Distance, Index = index.search(g[:100], k)