langchain_chroma.vectorstores.maximal_marginal_relevance

langchain_chroma.vectorstores.maximal_marginal_relevance(query_embedding: ndarray, embedding_list: list, lambda_mult: float = 0.5, k: int = 4) → List[int]

计算最大边际相关度。

参数

• query_embedding (ndarray) – 查询嵌入。

• embedding_list (list) – 从中选择嵌入的列表。

• lambda_mult (float) – 一个介于0和1之间的数字，用于确定结果之间的多样性程度，0对应最大多样性，1对应最小多样性。默认为0.5。

• k (int) – 要返回的文档数量。默认为4。

返回

通过最大边际相关度选择的嵌入的索引列表。

返回类型

List[int]