# Copyright (c) Facebook, Inc. and its affiliates. # # This source code is licensed under the MIT license found in the # LICENSE file in the root directory of this source tree. from __future__ import absolute_import, division, print_function # noqa E741 # translation of test_knn.lua import numpy as np import unittest import faiss from common import Randu10k, get_dataset_2, Randu10kUnbalanced ev = Randu10k() d = ev.d # Parameters inverted indexes ncentroids = int(4 * np.sqrt(ev.nb)) kprobe = int(np.sqrt(ncentroids)) # Parameters for LSH nbits = d # Parameters for indexes involving PQ M = int(d / 8) # for PQ: #subquantizers nbits_per_index = 8 # for PQ
# Copyright (c) Facebook, Inc. and its affiliates.
#
# This source code is licensed under the MIT license found in the
# LICENSE file in the root directory of this source tree.
#! /usr/bin/env python2
# translation of test_meta_index.lua
import numpy as np
import faiss
import unittest
from common import Randu10k
ru = Randu10k()
xb = ru.xb
xt = ru.xt
xq = ru.xq
nb, d = xb.shape
nq, d = xq.shape
class IDRemap(unittest.TestCase):
def test_id_remap_idmap(self):
# reference: index without remapping
index = faiss.IndexPQ(d, 8, 8)
k = 10
index.train(xt)
