Function search_documents

tests/test-tf-idf-ndcg.py:192–202

(
    collection,
    queries: List[Tuple[str, str]],
    top_k: int,
)

def search_documents(
    collection,
    queries: List[Tuple[str, str]],
    top_k: int,
) -> Dict[str, Dict[str, float]]:
    results = {}
    for query in tqdm(queries, desc="Searching vectors"):
        qid = query[0]
        query_results = search_document(collection, query[1], top_k)
        results[qid] = {doc_id: score for doc_id, score in query_results}
    return results


def batch_search_documents(collection, queries: List[str], top_k: int):