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

[skip ci] Update codes of benchmark (#11016) 

Signed-off-by: wangting0128 <ting.wang@zilliz.com>
pull/11020/head
wt 2021-11-01 18:06:03 +08:00 committed by GitHub
parent 556277616b
commit bb064e243f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 7 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -265,6 +265,13 @@ class AccAccuracyRunner(AccuracyRunner):
true_ids = case_param["true_ids"]
nq = case_metric.search["nq"]
top_k = case_metric.search["topk"]
start_time = time.time()
end_time = start_time + 500
cnt = 0
while cnt < 100 and start_time < end_time:
self.milvus.query(case_param["vector_query"], filter_query=case_param["filter_query"])
cnt += 1
start_time = time.time()
query_res = self.milvus.query(case_param["vector_query"], filter_query=case_param["filter_query"])
result_ids = self.milvus.get_ids(query_res)
# Calculate the accuracy of the result of query