milvus-io
/
milvus
mirror of https://github.com/milvus-io/milvus.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

[test]Update param of index and search for recall test (#22438) 

Signed-off-by: zhuwenxing <wenxing.zhu@zilliz.com>
pull/22459/head
zhuwenxing 2023-02-27 19:51:47 +08:00 committed by GitHub
parent 95f7a2d452
commit d6dcfe6fa9
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tests/python_client/deploy/scripts/first_recall_test.py
View File

@ -18,7 +18,7 @@ pymilvus_version = pymilvus.__version__
all_index_types = ["IVF_FLAT", "IVF_SQ8", "IVF_PQ", "HNSW", "ANNOY"]
default_index_params = [{"nlist": 128}, {"nlist": 128}, {"nlist": 128, "m": 16, "nbits": 8},
{"M": 48, "efConstruction": 500}, {"n_trees": 50}]
{"M": 48, "efConstruction": 100}, {"n_trees": 50}]
index_params_map = dict(zip(all_index_types, default_index_params))
@ -43,7 +43,7 @@ def gen_search_param(index_type, metric_type="L2"):
bin_search_params = {"metric_type": "HAMMING", "params": {"nprobe": nprobe}}
search_params.append(bin_search_params)
elif index_type in ["HNSW"]:
for ef in [200]:
for ef in [50]:
hnsw_search_param = {"metric_type": metric_type, "params": {"ef": ef}}
search_params.append(hnsw_search_param)
elif index_type == "ANNOY":

2
tests/python_client/deploy/scripts/second_recall_test.py
View File

@ -32,7 +32,7 @@ def gen_search_param(index_type, metric_type="L2"):
bin_search_params = {"metric_type": "HAMMING", "params": {"nprobe": nprobe}}
search_params.append(bin_search_params)
elif index_type in ["HNSW"]:
for ef in [200]:
for ef in [50]:
hnsw_search_param = {"metric_type": metric_type, "params": {"ef": ef}}
search_params.append(hnsw_search_param)
elif index_type == "ANNOY":