diff --git a/ann_benchmarks/.DS_Store b/ann_benchmarks/.DS_Store index 8c30ee3..ff82d8f 100644 Binary files a/ann_benchmarks/.DS_Store and b/ann_benchmarks/.DS_Store differ diff --git a/ann_benchmarks/algorithms/nmslib.py b/ann_benchmarks/algorithms/nmslib.py index 72f3ebc..8ee922f 100644 --- a/ann_benchmarks/algorithms/nmslib.py +++ b/ann_benchmarks/algorithms/nmslib.py @@ -66,6 +66,8 @@ def fit(self, X): X_trans = NmslibReuseIndex.matrToStrArray(csr_matrix(X)) else: X_trans = NmslibReuseIndex.intMatrToStrArray(X) + self._index = nmslib.init(space=self._nmslib_metric, method=self._method_name, data_type=nmslib.DataType.OBJECT_AS_STRING) + self._index.addDataPointBatch(X_trans) else: self._index = nmslib.init(space=self._nmslib_metric, method=self._method_name) self._index.addDataPointBatch(X)