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(self, cls)[source]
forward(self, 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(self)[source]
score(self, d)[source]
test(self, d)[source]