Lines Matching refs:faiss

8 import faiss
24 index = faiss.IndexFlatL2(d)
29 faiss.write_index(index, fname)
32 faiss.read_index(fname)
42 faiss.read_index(fname)
59 index = faiss.IndexFlatL2(d)
64 writer = faiss.PyCallbackIOWriter(f.write, 1234)
67 writer = faiss.BufferedIOWriter(writer, bsz)
69 faiss.write_index(index, writer)
77 index2 = faiss.deserialize_index(np.frombuffer(buf, dtype='uint8'))
81 faiss.vector_to_array(index.xb) == faiss.vector_to_array(index2.xb)
85 writer = faiss.PyCallbackIOWriter("blabla")
88 faiss.write_index, index, writer
100 reader = faiss.PyCallbackIOReader(f.read, 1234)
103 reader = faiss.BufferedIOReader(reader, bsz)
107 reader(faiss.swig_ptr(y), y.nbytes, 1)
117 index = faiss.IndexFlatL2(d)
123 faiss.write_index(index, fname)
126 reader = faiss.PyCallbackIOReader(f.read, 1234)
129 reader = faiss.BufferedIOReader(reader, bsz)
131 index2 = faiss.read_index(reader)
135 faiss.vector_to_array(index.xb),
136 faiss.vector_to_array(index2.xb)
140 reader = faiss.PyCallbackIOReader("blabla")
143 faiss.read_index, reader
166 index = faiss.IndexFlatL2(d)
172 faiss.write_index(index, fname)
174 reader = faiss.BufferedIOReader(
175 faiss.FileIOReader(fname), 1234)
177 index2 = faiss.read_index(reader)
181 faiss.vector_to_array(index.xb),
182 faiss.vector_to_array(index2.xb)
196 index = faiss.IndexFlatL2(d)
205 reader = faiss.PyCallbackIOReader(lambda size: os.read(rf, size))
206 return faiss.read_index(reader)
212 writer = faiss.PyCallbackIOWriter(lambda b: os.write(wf, b))
213 faiss.write_index(index, writer)
256 xq = faiss.randn((25, 10), 123)
257 xb = faiss.randn((25, 10), 124)
259 index = faiss.index_factory(10, fs)