admin/milvus
1
0
Fork 0
mirror of https://gitee.com/milvus-io/milvus.git synced 2025-12-28 22:45:26 +08:00
Code Issues Packages Projects Releases Wiki Activity

[skip ci] Add guarantee_timestamp of accuracy templet (#12773)

Browse Source 
Signed-off-by: wangting0128 <ting.wang@zilliz.com>
This commit is contained in:
wt 2021-12-06 18:12:58 +08:00 committed by GitHub
parent 32f3f78790
commit 7504dd5df2
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 7 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
tests/benchmark/milvus_benchmark/runners/accuracy.py
View File

@ -35,6 +35,7 @@ class AccuracyRunner(BaseRunner):
filter_query = []
top_ks = collection["top_ks"]
nqs = collection["nqs"]
guarantee_timestamp = collection["guarantee_timestamp"] if "guarantee_timestamp" in collection else None
search_params = collection["search_params"]
search_params = utils.generate_combinations(search_params)
cases = list()
@ -67,7 +68,8 @@ class AccuracyRunner(BaseRunner):
"nq": nq,
"topk": top_k,
"search_param": search_param,
"filter": filter_param
"filter": filter_param,
"guarantee_timestamp": guarantee_timestamp
}
vector_query = {"vector": {index_field_name: search_info}}
case = {
@ -79,7 +81,8 @@ class AccuracyRunner(BaseRunner):
"vector_type": vector_type,
"collection_size": collection_size,
"filter_query": filter_query,
"vector_query": vector_query
"vector_query": vector_query,
"guarantee_timestamp": guarantee_timestamp
}
cases.append(case)
case_metrics.append(case_metric)
@ -96,7 +99,8 @@ class AccuracyRunner(BaseRunner):
collection_size = case_param["collection_size"]
nq = case_metric.search["nq"]
top_k = case_metric.search["topk"]
query_res = self.milvus.query(case_param["vector_query"], filter_query=case_param["filter_query"])
query_res = self.milvus.query(case_param["vector_query"], filter_query=case_param["filter_query"],
guarantee_timestamp=case_param["guarantee_timestamp"])
true_ids = utils.get_ground_truth_ids(collection_size)
logger.debug({"true_ids": [len(true_ids[0]), len(true_ids[0])]})
result_ids = self.milvus.get_ids(query_res)