primeqa.ir.dense.dpr_top.dpr.config.DPRSearchArguments

class primeqa.ir.dense.dpr_top.dpr.config.DPRSearchArguments(output_dir: str = '', queries: str = '', bsize: int = 10, index_location: Optional[str] = None, model_name_or_path: str = '', qry_encoder_name_or_path: str = 'facebook/dpr-question_encoder-multiset-base', ctx_encoder_name_or_path: str = 'facebook/dpr-ctx_encoder-multiset-base', top_k: int = 10, rescore_only: bool = False, max_doc_length: int = 128)

Bases: object

Arguments used in search

Methods

Attributes

bsize
ctx_encoder_name_or_path
index_location
max_doc_length
model_name_or_path
output_dir
qry_encoder_name_or_path
queries
rescore_only
top_k