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

Correctly set number of probes in FAISS.
erikbern · May 3, 2018 · 94da489 · 94da489
2 parents 9bb93fe + b956147
commit 94da489
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':