diff --git a/ann_benchmarks/algorithms/hnswlib/module.py b/ann_benchmarks/algorithms/hnswlib/module.py index 92c7bed93..d70d56117 100644 --- a/ann_benchmarks/algorithms/hnswlib/module.py +++ b/ann_benchmarks/algorithms/hnswlib/module.py @@ -10,7 +10,7 @@ def __init__(self, metric, method_param): self.method_param = method_param # print(self.method_param,save_index,query_param) # self.ef=query_param['ef'] - self.name = "hnswlib (%s)" % (self.method_param) + # self.name = "hnswlib (%s)" % (self.method_param) def fit(self, X): # Only l2 is supported currently @@ -24,6 +24,7 @@ def fit(self, X): def set_query_arguments(self, ef): self.p.set_ef(ef) + self.name = "hnswlib (%s, 'efQuery': %s)" % (self.method_param, ef) def query(self, v, n): # print(np.expand_dims(v,axis=0).shape)