Correctly set number of probes in FAISS.

yurymalkov · May 3, 2018 · b6839f2 · b6839f2
1 parent 059b202
commit b6839f2
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':