Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
BaiXuePrincess
milvus
提交
905722bd
milvus
项目概览
BaiXuePrincess
/
milvus
与 Fork 源项目一致
从无法访问的项目Fork
通知
7
Star
4
Fork
0
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
0
列表
看板
标记
里程碑
合并请求
0
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
milvus
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
0
Issue
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
体验新版 GitCode,发现更多精彩内容 >>
提交
905722bd
编写于
4月 30, 2019
作者:
G
groot
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
add python sample
Former-commit-id: 2c0825ad6ed774eb5ad1999b1439f840dc602bf1
上级
ae5e6720
变更
1
显示空白变更内容
内联
并排
Showing
1 changed file
with
89 addition
and
0 deletion
+89
-0
cpp/src/thrift/gen-py/py_sample.py
cpp/src/thrift/gen-py/py_sample.py
+89
-0
未找到文件。
cpp/src/thrift/gen-py/py_sample.py
0 → 100644
浏览文件 @
905722bd
import
time
import
struct
from
test_proj_env.zilliz
import
VecService
from
thrift
import
Thrift
from
thrift.transport
import
TSocket
from
thrift.transport
import
TTransport
from
thrift.protocol
import
TBinaryProtocol
,
TCompactProtocol
,
TJSONProtocol
def
print_time_cost
(
desc
,
t
):
time_now
=
time
.
time
()
print
(
desc
+
' cost '
,
time_now
-
t
,
' sec'
)
return
time_now
def
test_vecwise
():
try
:
time_start
=
time
.
time
()
#connect
transport
=
TSocket
.
TSocket
(
'localhost'
,
33001
)
transport
=
TTransport
.
TBufferedTransport
(
transport
)
protocol
=
TJSONProtocol
.
TJSONProtocol
(
transport
)
client
=
VecService
.
Client
(
protocol
)
transport
.
open
()
time_start
=
print_time_cost
(
'connection'
,
time_start
)
#add group
group
=
VecService
.
VecGroup
(
"py_group_"
+
time
.
strftime
(
'%H%M%S'
),
256
)
client
.
add_group
(
group
)
time_start
=
print_time_cost
(
'add group'
,
time_start
)
#build vectors
# vec_list = VecService.VecTensorList([])
# for i in range(10000):
# vec = VecService.VecTensor("normal_"+str(i), [])
# for k in range(group.dimension):
# vec.tensor.append(k)
# vec_list.tensor_list.append(vec)
#time_start = print_time_cost('build normal vectors', time_start)
#add vectors
#client.add_vector_batch(group.id, vec_list)
#time_start = print_time_cost('add normal vectors', time_start))
# build binary vectors
bin_vec_list
=
VecService
.
VecBinaryTensorList
([])
for
i
in
range
(
10000
):
a
=
[]
for
k
in
range
(
group
.
dimension
):
a
.
append
(
i
+
k
)
bin_vec
=
VecService
.
VecBinaryTensor
(
"binary_"
+
str
(
i
),
bytes
())
bin_vec
.
tensor
=
struct
.
pack
(
str
(
group
.
dimension
)
+
"d"
,
*
a
)
bin_vec_list
.
tensor_list
.
append
(
bin_vec
)
time_start
=
print_time_cost
(
'build binary vectors'
,
time_start
)
# add vectors
client
.
add_binary_vector_batch
(
group
.
id
,
bin_vec_list
)
time_start
=
print_time_cost
(
'add binary vectors'
,
time_start
)
time
.
sleep
(
5
)
time_start
=
print_time_cost
(
'sleep 5 seconds'
,
time_start
)
# search vector
a
=
[]
for
k
in
range
(
group
.
dimension
):
a
.
append
(
300
+
k
)
bin_vec
=
VecService
.
VecBinaryTensor
(
"binary_search"
,
bytes
())
bin_vec
.
tensor
=
struct
.
pack
(
str
(
group
.
dimension
)
+
"d"
,
*
a
)
filter
=
VecService
.
VecSearchFilter
()
res
=
VecService
.
VecSearchResult
()
res
=
client
.
search_binary_vector
(
group
.
id
,
5
,
bin_vec
,
filter
)
time_start
=
print_time_cost
(
'search binary vectors'
,
time_start
)
print
(
'result count: '
+
str
(
len
(
res
.
result_list
)))
for
item
in
res
.
result_list
:
print
(
item
.
uid
)
transport
.
close
()
time_start
=
print_time_cost
(
'close connection'
,
time_start
)
except
Thrift
.
TException
as
ex
:
print
(
ex
.
message
)
test_vecwise
()
\ No newline at end of file
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录