未验证 提交 d6dcfe6f 编写于 作者: Z zhuwenxing 提交者: GitHub

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

Signed-off-by: Nzhuwenxing <wenxing.zhu@zilliz.com>
上级 95f7a2d4
......@@ -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":
......
......@@ -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":
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册