向量数据库的使用

向量数据库

向量数据库是一种专门用于存储、管理和查询高维向量数据的数据库系统。随着人工智能和机器学习的广泛应用,向量数据库在处理非结构化数据(如文本、图像、音频和视频等)的任务中变得越来越重要。本文将介绍快速使用Chroma

安装

安装chromadb向量数据库

复制代码
pip install chromadb 

创建链接

创建客户端

复制代码
import chromadb
chroma_client = chromadb.Client()

创建集合

Chroma数据结构,包括集合、文档和Embedding。

复制代码
collection = chroma_client.create_collection(name="my_collection")

添加文档

添加文档到集合中

复制代码
collection.add(
    documents=[
        "This is a document about pineapple",
        "This is a document about oranges"
    ],
    ids=["id1", "id2"]
)

搜索

搜索文档并指定返回文档数

复制代码
results = collection.query(
    query_texts=["This is a query document about hawaii"], # Chroma will embed this for you
    n_results=2 # how many results to return
)
print(results)

查看结果

可以看到夏威夷和菠萝更相似。

复制代码
{
  'documents': [[
      'This is a document about pineapple',
      'This is a document about oranges'
  ]],
  'ids': [['id1', 'id2']],
  'distances': [[1.0404009819030762, 1.243080496788025]],
  'uris': None,
  'data': None,
  'metadatas': [[None, None]],
  'embeddings': None,
}

总结

向量数据库是 RAG 中的重要组件之一,文档索引会存储在向量数据库中,随着大模型的流行,感觉向量数据库也会持续发展,进一步提高性能。

相关推荐
玩电脑的辣条哥3 天前
向量数据库是什么,它有什么作用?
数据库·向量数据库
重整旗鼓~5 天前
3.milvus索引-HNSW
milvus·向量数据库
重整旗鼓~8 天前
4.milvus索引FLAT
milvus·向量数据库
无极低码9 天前
基于deepseek的智能语音客服【第四讲】封装milvus数据库连接池封装
数据库·人工智能·milvus·连接池·向量数据库·向量化
重整旗鼓~13 天前
向量数据库milvus
人工智能·算法·机器学习·向量数据库
韩曙亮15 天前
【AI 大模型】RAG 检索增强生成 ⑤ ( 向量数据库 | 向量数据库 索引结构和搜索算法 | 常见 向量数据库 对比 | 安装并使用 向量数据库 chromadb 案例 )
数据库·人工智能·大模型·openai·向量数据库·ai大模型·chromadb
kse_music23 天前
如何选择开源向量数据库
数据库·开源·向量数据库
探模之翼1 个月前
基于Linux环境部署和使用ElasticSearch搜索引擎
linux·elasticsearch·搜索引擎·向量数据库·多模态检索
山塘小鱼儿1 个月前
Milvus高性能向量数据库与大模型结合
数据库·阿里云·向量数据库
oscar9992 个月前
AI向量数据库之LanceDB快速介绍
数据库·人工智能·向量数据库·lancedb