admin/milvus
1
0
Fork 0
mirror of https://gitee.com/milvus-io/milvus.git synced 2026-01-07 19:31:51 +08:00
Code Issues Packages Projects Releases Wiki Activity

update

Browse Source 
Signed-off-by: zhuwenxing <wenxing.zhu@zilliz.com>
This commit is contained in:
zhuwenxing 2024-07-11 18:50:55 +08:00
parent a76241932a
commit 6ac2922212
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tests/python_client/filter_search/prepare_custom_data.py
View File

@ -45,6 +45,8 @@ def prepare_data(host="127.0.0.1", port=19530, minio_host="127.0.0.1", data_size
test_emb_matrix = test_emb_matrix[:test_data_size]
train_emb_matrix = pd.read_parquet(f"{data_dir}/train-00-of-10.parquet")["emb"].values
if data_size > len(train_emb_matrix):
train_emb_matrix = np.concatenate([train_emb_matrix] * (data_size // len(train_emb_matrix) + 1), axis=0)
train_emb_matrix = train_emb_matrix[:data_size]
#
# # generate data