Set KnnDescriptor.desc_name in the Benchmarking core framework in FAISS like other descriptors (#4109)

Summary:
Pull Request resolved: https://github.com/facebookresearch/faiss/pull/4109

Set KnnDescriptor.desc_name in the Benchmarking core framework in FAISS like other descriptors

Reviewed By: mnorris11

Differential Revision: D67539874

fbshipit-source-id: 09ffb76296f466ae2d3b0eb551917f429bc7300f
quiet_install
Satyendra Mishra 2024-12-23 11:27:51 -08:00 committed by Facebook GitHub Bot
parent ab8cb9cc20
commit 3c8dc41949
1 changed files with 3 additions and 0 deletions

View File

@ -341,6 +341,8 @@ class KnnDescriptor(IndexBaseDescriptor):
return hash(str(self))
def get_name(self):
if self.desc_name is not None:
return self.desc_name
name = self.index_desc.get_name()
name += IndexBaseDescriptor.param_dict_to_name(self.search_params)
name += self.query_dataset.get_filename(KnnDescriptor.FILENAME_PREFIX)
@ -350,6 +352,7 @@ class KnnDescriptor(IndexBaseDescriptor):
name += "rec."
else:
name += "knn."
self.desc_name = name
return name
def flat_name(self):