最大边际相关性#

langchain_databricks.utils.maximal_marginal_relevance(query_embedding: ndarray, embedding_list: list, lambda_mult: float = 0.5, k: int = 4) → List[int][source]#

计算最大边际相关性。

Parameters:

  • query_embedding (ndarray) – 查询嵌入。

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

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

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

Returns:

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

Return type:

列表[int]