def test_OIVFPQ(self):
# Parameters inverted indexes
ncentroids = 50
M = 4
ev = Randu10kUnbalanced()
d = ev.d
quantizer = faiss.IndexFlatL2(d)
index = faiss.IndexIVFPQ(quantizer, d, ncentroids, M, 8)
index.nprobe = 5
res = ev.launch('IVFPQ', index)
e_ivfpq = ev.evalres(res)
quantizer = faiss.IndexFlatL2(d)
index_ivfpq = faiss.IndexIVFPQ(quantizer, d, ncentroids, M, 8)
index_ivfpq.nprobe = 5
opq_matrix = faiss.OPQMatrix(d, M)
opq_matrix.niter = 10
index = faiss.IndexPreTransform(opq_matrix, index_ivfpq)
res = ev.launch('O+IVFPQ', index)
e_oivfpq = ev.evalres(res)
# verify same on OIVFPQ
for r in 1, 10, 100:
print(e_oivfpq[r], e_ivfpq[r])
assert(e_oivfpq[r] >= e_ivfpq[r])
def test_OPQ(self):
M = 4
ev = Randu10kUnbalanced()
d = ev.d
index = faiss.IndexPQ(d, M, 8)
res = ev.launch('PQ', index)
e_pq = ev.evalres(res)
index_pq = faiss.IndexPQ(d, M, 8)
opq_matrix = faiss.OPQMatrix(d, M)
# opq_matrix.verbose = true
opq_matrix.niter = 10
opq_matrix.niter_pq = 4
index = faiss.IndexPreTransform(opq_matrix, index_pq)
res = ev.launch('OPQ', index)
e_opq = ev.evalres(res)
print('e_pq=%s' % e_pq)
print('e_opq=%s' % e_opq)
# verify that OPQ better than PQ
for r in 1, 10, 100:
assert(e_opq[r] > e_pq[r])
