semantic_router.linear - Aurelio AI

similarity_matrix

def similarity_matrix(xq: np.ndarray, index: np.ndarray) -> np.ndarray

Compute the similarity scores between a query vector and a set of vectors. Arguments:

xq: A query vector (1d ndarray)
index: A set of vectors.

Returns: np.ndarray: The similarity between the query vector and the set of vectors.

top_scores

def top_scores(sim: np.ndarray,
               top_k: int = 5) -> Tuple[np.ndarray, np.ndarray]

Get the top scores and indices from a similarity matrix. Arguments:

sim: A similarity matrix.
top_k: The number of top scores to get.

Returns: Tuple[np.ndarray, np.ndarray]: The top scores and indices.

Quickstart route

⌘I