提示信息

Milvus 用法?

佳薪 已关注

Milvus是一个开源的向量搜索引擎,提供了可插拔、高度可扩展的数据处理方式和多种索引类型。下面是一个使用Python的Milvus基本示例:

  1. 首先安装Python Milvus SDK:
  1. pip install pymilvus==2.0.0rc5

注意: 请根据你的环境选择合适的版本,以上命令可能会随着时间推移而变化。

  1. 启动Milvus服务器。您可以参照官方文档关于如何通过Docker启动Milvus。

  2. 使用Python连接到Milvus并创建向量数据库:

  1. from pymilvus import connections, FieldSchema, CollectionSchema, DataType, Collection
  2. # 连接到Milvus服务器
  3. connections.connect("default", host="localhost", port="19530")
  4. # 定义字段和集合模式
  5. dim = 128
  6. fields = [
  7. FieldSchema(name="id", dtype=DataType.INT64, is_primary=True),
  8. FieldSchema(name="embedding", dtype=DataType.FLOAT_VECTOR, dim=dim)
  9. ]
  10. schema = CollectionSchema(fields=fields, description="test collection")
  11. # 创建向量数据库
  12. collection = Collection(name="my_collection", schema=schema)
  13. # 插入一些向量数据
  14. import numpy as np
  15. vectors = np.random.rand(100, dim).tolist()
  16. ids = [i for i in range(len(vectors))]
  17. mr = collection.insert([ids, vectors])
  18. # 执行查询
  19. topK = 10
  20. query_embedding = vectors[0]
  21. res = collection.search(query_embedding, "embedding", param={'nprobe': 10}, limit=topK)

以上代码连接到本地的Milvus服务器,创建一个集合(相当于传统数据库中的表),并插入一些随机生成的向量数据。然后执行一个查询,返回查询向量的最近邻。

    遇到难题? "AI大模型GPT4.0、GPT" 是你的私人解答专家! 点击按钮去提问......
佳薪 关注 已关注

最近一次登录:2024-06-17 10:04:40   

暂时还没有签名,请关注我或评论我的文章
×
免费图表工具,画流程图、架构图