From 6cac346835b845a078cee4e079f3ae8e8b6f36fc Mon Sep 17 00:00:00 2001 From: wt Date: Wed, 20 Oct 2021 20:32:36 +0800 Subject: [PATCH] [skip ci] Update parse_ann_collection_name func on benchmark (#10267) Signed-off-by: wangting0128 --- tests/benchmark/milvus_benchmark/parser.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/tests/benchmark/milvus_benchmark/parser.py b/tests/benchmark/milvus_benchmark/parser.py index 0337b1e4c5..28abe6ceb2 100644 --- a/tests/benchmark/milvus_benchmark/parser.py +++ b/tests/benchmark/milvus_benchmark/parser.py @@ -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):