primeqa.ir.dense.colbert_top.colbert.search.candidate_generation.CandidateGeneration#

class primeqa.ir.dense.colbert_top.colbert.search.candidate_generation.CandidateGeneration(use_gpu=True)#

Bases: object

Methods

generate_candidate_eids

generate_candidate_pids

generate_candidate_scores

generate_candidates

get_cells