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 parse_ann_collection_name func on benchmark (#10267) 

Signed-off-by: wangting0128 <ting.wang@zilliz.com>
pull/10316/head
wt 2021-10-20 20:32:36 +08:00 committed by GitHub
parent 3610fe89c0
commit 6cac346835
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
tests/benchmark/milvus_benchmark/parser.py
View File

@ -50,15 +50,16 @@ def parse_ann_collection_name(collection_name):
metric = collection_name.split("_")[2]
# metric = collection_name.attrs['distance']
# dimension = len(collection_name["train"][0])
metric_type = ''
if metric == "euclidean":
metric_type = "l2"
elif metric == "angular":
elif metric == "angular":
metric_type = "ip"
elif metric == "jaccard":
elif metric == "jaccard":
metric_type = "jaccard"
elif metric == "hamming":
metric_type = "hamming"
return (data_type, dimension, metric_type)
return data_type, dimension, metric_type
def search_params_parser(param):