Merge pull request #3 from maumueller/build-parameters-1

Correctly set number of probes in FAISS.
yurymalkov · May 3, 2018 · 3925d4d · 3925d4d
2 parents 059b202 + b6839f2
commit 3925d4d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ann_benchmarks/algorithms/faiss.py b/ann_benchmarks/algorithms/faiss.py
@@ -57,7 +57,7 @@ def fit(self, X):
 
     def set_query_arguments(self, n_probe):
         self._n_probe = n_probe
-        self._index.n_probe = self._n_probe
+        self._index.nprobe = self._n_probe
 
     def query(self, v, n):
         if self._metric == 'angular':