python - 如何在“add_with_ids”训练索引上返回“faiss”唯一向量 id？

Question

我正在使用 Facebook 的 faiss 索引和使用该add_with_ids方法的自定义索引。在推理时，我使用distance, ID = model.search()它返回训练时使用的自定义 ID。是否可以在不重新培训的情况下也返回一个唯一的 ID？还是只返回实际最接近的向量？

谢谢！

标签： pythonknn

Answer 1

由于您提供了实际的向量，因此您大概知道如何将 id 映射到向量。大多数 Faiss 索引不存储向量，因为它们需要被压缩以适应 RAM。