faiss

Efficient similarity search and clustering of dense vectors

brewmacoslinux
Try with needOr install directly
Source

About

Efficient similarity search and clustering of dense vectors

Commands

faiss

Examples

Build a flat index and search for similar vectors$ python -c "import faiss; index = faiss.IndexFlatL2(128); index.add(vectors); distances, indices = index.search(query_vectors, k=10)"
Create an IVF index for faster approximate nearest neighbor search$ python -c "import faiss; quantizer = faiss.IndexFlatL2(128); index = faiss.IndexIVFFlat(quantizer, 128, 100); index.train(train_vectors); index.add(vectors)"
Save and load an index for persistence$ python -c "import faiss; faiss.write_index(index, 'index.faiss'); loaded_index = faiss.read_index('index.faiss')"