我正在使用 FAISS 库为大型数据语料库构建示意图搜索模型。对于搜索操作,我使用 FAISS 库中的“IndexFlatL2”方法,该方法在后台使用欧几里德距离来查找最近的邻居。python实现如图所示。
import faiss as fs
fs_index = fs.IndexFlatL2(768)
fs_index.add(train_embeddings)
D,I = fs_index.search(query,k=1)
现在,我试图将 fs_index 对象存储为模型文件(即像传统的 ML 模型 pickle 或 joblib 文件),以便我可以使用相同的文件来预测其他模块/其他包中的其他查询。
那么,有没有办法将此索引对象存储为 joblib/pickle 文件,以便根据需要在任何其他地方进一步使用此模型?
提前致谢