admin/milvus
1
0
Fork 0
mirror of https://gitee.com/milvus-io/milvus.git synced 2026-01-07 19:31:51 +08:00
Code Issues Packages Projects Releases Wiki Activity

add back nsg

Browse Source 
Signed-off-by: Xiaohai Xu <xiaohaix@student.unimelb.edu.au>
This commit is contained in:
Xiaohai Xu 2020-03-10 15:15:19 +08:00
parent 76cf09c8fa
commit a3b2842935
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
tests/milvus_python_test/utils.py
View File

@ -588,7 +588,7 @@ def gen_simple_index():
IndexType.IVF_SQ8H,
IndexType.IVF_PQ,
IndexType.HNSW,
# IndexType.RNSG
IndexType.RNSG
]
params = [
{"nlist": 1024},
@ -597,7 +597,7 @@ def gen_simple_index():
{"nlist": 1024},
{"nlist": 1024, "m": 16},
{"M": 16, "efConstruction": 500},
# {"search_length": 100, "out_degree": 40, "candidate_pool_size": 66, "knng": 100}
{"search_length": 50, "out_degree": 40, "candidate_pool_size": 100, "knng": 50}
]
index_params = []
@ -611,8 +611,8 @@ def get_search_param(index_type):
return {"nprobe": 32}
elif index_type == IndexType.HNSW:
return {"ef": 64}
# elif index_type == IndexType.RNSG:
# return {"search_length": 100}
elif index_type == IndexType.RNSG:
return {"search_length": 50}
else:
logging.getLogger().info("Invalid index_type.")