Chroma向量数据库支持的距离计算公式_chroma db 自定义距离

作者：不正经 | 2024-02-27 12:41:56

踩

chroma db 自定义距离

Chroma DB默认采用squared L2 norm作为衡量两个向量距离的计算方式，可以在创建collection的时候更改metadata的hnsw:space字段指定其他距离计算公式，方式如下：

 collection = client.create_collection(
        name="collection_name",
        metadata={"hnsw:space": "cosine"} # l2 is the default
)
1
2
3
4

支持的全部计算公式，将hnsw:space字段设置为parameter一栏的标识符即可设置

Distance	parameter	Equation
Squared L2	‘l2’	$\sum\left(A_i-B_i\right)^2$
Inner product	‘ip’	$\sum\left(A_i \times B_i\right)$
Cosine similarity	‘cosine’	$\frac{\sum\left(A_i \times B_i\right)}{\sqrt{\sum\left(A_i^2\right)} \cdot \sqrt{\sum\left(B_i^2\right)}}$

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/不正经/article/detail/153026