用于milvus的python sdk;alpha版本
pymilvus-test的Python项目详细描述
milvus python sdk——pymilvus
为milvus使用milvus python sdk 下载
pymilvus只支持python >= 3.4
,在3.4、3.5、3.6、3.7下进行了全面测试。
pymilvus可以通过pip
下载。如果没有用,请尝试pip3
$ pip install pymilvus
相应地支持不同版本的milvus和最低/最高pymilvus版本
Milvus version | Lowest pymilvus version supported | Highest pymivus version supported |
---|---|---|
0.3.0 | - | 0.1.13 |
0.3.1 | 0.1.14 | 0.1.25 |
0.4.0 | 0.2.0 | - |
您可以通过以下方式下载特定版本:
$ pip install pymilvus==0.1.13
如果要将pymilvus
升级到最新版本
$ pip install --upgrade pymilvus
导入
frommilvusimportMilvus,IndexType,Status
开始
初始化一个Milvus
实例并connect
到服务器
>>>milvus=Milvus()>>>milvus.connect(host='SERVER-HOST',port='SERVER-PORT')Status(code=0,message="Success")
成功连接后,您可以获得服务器的版本
>>>milvus.server_version()(Status(code=0,message='Success'),0.3.1)# this is example version, the real version may vary
添加新的table
第一组参数
>>>param={'table_name':'test01','dimension':256,'index_type':IndexType.FLAT,'store_raw_vector':False}
然后create table
>>>milvus.create_table(param)Status(message='Table test01 created!',code=0)
描述一下我们刚创建的表格
>>>milvus.describe_table('test01')(Status(code=0,message='Success!'),TableSchema(table_name='test01',dimension=256,index_type=<IndexType:FLAT>,store_raw_vector=False))
将向量添加到表test01
首先创建20个256维向量。
- 请注意,我们在这里使用的
random
和pprint
用于创建伪向量数据和漂亮的打印,您的项目中可能不需要它们
>>>importrandom>>>frompprintimportpprint>>>dim=256# Dimension of the vector# Initialize 20 vectors of 256-dimension>>>fake_vectors=[[random.random()for_inrange(dim)]for_inrange(20)]
然后将向量添加到表test01
>>>status,ids=milvus.add_vectors(table_name='test01',records=vectors)>>>print(status)Status(code=0,message='Success')>>>pprint(ids)# List of ids returned2345532113551123312245748929023489...
搜索向量
# create 5 vectors of 256-dimension>>>q_records=[[random.random()for_inrange(dim)]for_inrange(5)]
然后获取结果
>>>status,results=milvus.search_vectors(table_name='test01',query_records=q_records,top_k=10)>>>print(status)Status(code=0,message='Success')>>>pprint(results)# Searched top_k vectors
删除刚创建的表
>>>milvus.delete_table(table_name='test01')Status(code=0,message='Success')
断开与服务器的连接
>>>milvus.disconnect()Status(code=0,message='Success')
示例python
在examples/
中有一些小例子,您可以在那里找到更多指南。
生成文档
$ sphinx-build -b html doc/en/ doc/en/build
如果遇到任何问题或错误,请打开新问题