Source code for purrfectmeow.tc04_kmn.base

import numpy

from purrfectmeow.meow.chaus import SimilarityResult
from purrfectmeow.meow.felis import Document

from .cosine import CosineSim


[docs] class KhaoManee:
[docs] @classmethod def searching( cls, query_embed: numpy.ndarray, sentence_embed: numpy.ndarray | list[numpy.ndarray], documents: list[Document], top_k: int, ) -> list[SimilarityResult]: return CosineSim.vector_search(query_embed, sentence_embed, documents, top_k)