capreolus.reranker.ptparade

Module Contents

Classes

PTParade_Class

PTParade

PyTorch implementation of PARADE.

class capreolus.reranker.ptparade.PTParade_Class(extractor, config, *args, **kwargs)[source]

Bases: torch.nn.Module

aggregate_using_transformer(cls)[source]
forward(doc_input, doc_mask, doc_seg)[source]
class capreolus.reranker.ptparade.PTParade(config=None, provide=None, share_dependency_objects=False, build=True)[source]

Bases: capreolus.reranker.Reranker

PyTorch implementation of PARADE.

PARADE: Passage Representation Aggregation for Document Reranking. Canjia Li, Andrew Yates, Sean MacAvaney, Ben He, and Yingfei Sun. arXiv 2020. https://arxiv.org/pdf/2008.09093.pdf

module_name = 'ptparade'[source]
dependencies[source]
config_spec[source]
build_model()[source]
score(d)[source]
test(d)[source]