langchain_aws.utilities.math.cosine_similarity_top_k

langchain_aws.utilities.math.cosine_similarity_top_k(X: Union[List[List[float]], List[ndarray], ndarray], Y: Union[List[List[float]], List[ndarray], ndarray], top_k: Optional[int] = 5, score_threshold: Optional[float] = None) → Tuple[List[Tuple[int, int]], List[float]]

Optionally with top-k and score threshold, row-wise cosine similarity.

参数

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

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

• top_k (Optional[int]) – 返回的最大结果数。

• score_threshold (Optional[float]) – 结果的最小余弦相似度。

返回

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

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

返回类型

Tuple[List[Tuple[int, int]], List[float]]