Chroma 教程

内置嵌入函数一览

🎉摘要:详细介绍chromadb.utils.embedding_functions下所有可用的嵌入函数,包含DefaultEmbeddingFunction、OpenAIEmbeddingFunction、OllamaEmbeddingFunction等完整列表及对应服务说明,并提供代码示例查看当前版本支持的嵌入函数。

在 chromadb.utils.embedding_functions 下面有一大堆现成的嵌入函数,如下图:

或者直接查看 _init_.py 文件,给出所有可以使用的嵌入函数。

下面表格对这些嵌入函数简单介绍:

类名对应服务/模型
DefaultEmbeddingFunctionall-MiniLM-L6-v2(本地 ONNX)
SentenceTransformerEmbeddingFunction任意 sentence-transformers 模型(本地)
HuggingFaceEmbeddingFunctionHF Inference API
HuggingFaceSparseEmbeddingFunctionHF 稀疏向量
OpenAIEmbeddingFunctionOpenAI text-embedding-*
GoogleGenerativeAiEmbeddingFunction GoogleGeminiEmbeddingFunctionGoogleVertexEmbeddingFunction谷歌系列
CohereEmbeddingFunctionCohere
JinaEmbeddingFunctionJina AI
MistralEmbeddingFunctionMistral
VoyageAIEmbeddingFunctionVoyage AI
NomicEmbeddingFunctionNomic
TogetherAIEmbeddingFunctionTogether AI
PerplexityEmbeddingFunctionPerplexity
MorphEmbeddingFunctionMorph
BasetenEmbeddingFunctionBaseten
CloudflareWorkersAIEmbeddingFunctionCloudflare Workers AI
AmazonBedrockEmbeddingFunctionAWS Bedrock
OllamaEmbeddingFunction本地 Ollama
InstructorEmbeddingFunctionInstructor 系列
Text2VecEmbeddingFunctiontext2vec(中文常用)
OpenCLIPEmbeddingFunction图文多模态
RoboflowEmbeddingFunctionRoboflow
Bm25EmbeddingFunctionChromaBm25EmbeddingFunctionBM25 稀疏
FastembedSparseEmbeddingFunctionChromaCloudSpladeEmbeddingFunction ChromaCloudQwenEmbeddingFunction ChromaLangchainEmbeddingFunction稀疏向量 / Cloud 专用

注意了,不同版本 Chroma 这个列表会变,想看自己装的版本有哪些嵌入函数,使用下面代码片段:

import chromadb.utils.embedding_functions as ef

names = [n for n in dir(ef) if n.endswith("EmbeddingFunction")]
print(len(names))

for n in names:
    print(" ", n)

记住这些嵌入函数的位置,后续如果自己要实现自己的嵌入函数,直接拷贝一个,模仿编写就可以了。

  

说说我的看法
全部评论()
没有评论
关于
本网站专注于 Java、数据库(MySQL、Oracle)、Linux、软件架构及大数据等多领域技术知识分享。涵盖丰富的原创与精选技术文章,助力技术传播与交流。无论是技术新手渴望入门,还是资深开发者寻求进阶,这里都能为您提供深度见解与实用经验,让复杂编码变得轻松易懂,携手共赴技术提升新高度。如有侵权,请来信告知:hxstrive@outlook.com
其他应用
公众号