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