提交 a3b28429 编写于 作者: X Xiaohai Xu

add back nsg

Signed-off-by: NXiaohai Xu <xiaohaix@student.unimelb.edu.au>
上级 76cf09c8
...@@ -588,7 +588,7 @@ def gen_simple_index(): ...@@ -588,7 +588,7 @@ def gen_simple_index():
IndexType.IVF_SQ8H, IndexType.IVF_SQ8H,
IndexType.IVF_PQ, IndexType.IVF_PQ,
IndexType.HNSW, IndexType.HNSW,
# IndexType.RNSG IndexType.RNSG
] ]
params = [ params = [
{"nlist": 1024}, {"nlist": 1024},
...@@ -597,7 +597,7 @@ def gen_simple_index(): ...@@ -597,7 +597,7 @@ def gen_simple_index():
{"nlist": 1024}, {"nlist": 1024},
{"nlist": 1024, "m": 16}, {"nlist": 1024, "m": 16},
{"M": 16, "efConstruction": 500}, {"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 = [] index_params = []
...@@ -611,8 +611,8 @@ def get_search_param(index_type): ...@@ -611,8 +611,8 @@ def get_search_param(index_type):
return {"nprobe": 32} return {"nprobe": 32}
elif index_type == IndexType.HNSW: elif index_type == IndexType.HNSW:
return {"ef": 64} return {"ef": 64}
# elif index_type == IndexType.RNSG: elif index_type == IndexType.RNSG:
# return {"search_length": 100} return {"search_length": 50}
else: else:
logging.getLogger().info("Invalid index_type.") logging.getLogger().info("Invalid index_type.")
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册