def run_retrieval()

in evaluation_pipeline/retrieval.py [0:0]

• 10 lines of code
• 3 McCabe index (conditional complexity)

def run_retrieval(fe, query_ids, db, model_name, threshold, k):
    # loop through each query
    retreival_dict = {}
    query_lookup = {}
    for query in query_ids.keys():
        # for later identifying query
        query_id = query_ids[query]
        query_lookup[query_id] = query
        # perform retrieval
        results = query_and_result(fe, query, db=db,
        model_name=model_name, threshold=threshold, k=k)
        retreival_dict[query_id] = results
    return retreival_dict, query_lookup