cosine_similarity_top_k

langchain_community.utils.math.cosine_similarity_top_k(X: List[List[float]] | List[ndarray] | ndarray, Y: List[List[float]] | List[ndarray] | ndarray, top_k: int | None = 5, score_threshold: float | None = None) → Tuple[List[Tuple[int, int]], List[float]]

行间余弦相似度，带有可选的 top-k 和分数阈值过滤。

Parameters:

• X (List[List[float]] | List[ndarray] | ndarray) – 矩阵。

• Y (List[List[float]] | List[ndarray] | ndarray) – 矩阵，与X的宽度相同。

• top_k (int | None) – 返回的最大结果数。

• score_threshold (float | None) – 结果的最小余弦相似度。

Returns:

两个列表的元组。第一个包含索引的两元组（X_idx, Y_idx），

第二个包含相应的余弦相似度。

Return type:

元组[列表[元组[int, int]], 列表[float]]