capreolus.reranker.TFCEDRKNRM

Module Contents

Classes

TFCEDRKNRM_Class
TFCEDRKNRM	TensorFlow implementation of CEDR-KNRM.

capreolus.reranker.TFCEDRKNRM.logger

class capreolus.reranker.TFCEDRKNRM.TFCEDRKNRM_Class(extractor, config, *args, **kwargs)

Bases: tensorflow.keras.layers.Layer

masked_simmats(self, embeddings, bert_mask, bert_segments)

knrm(self, bert_output, bert_mask, bert_segments, batch_size)

call(self, x, **kwargs)

predict_step(self, data)

score(self, x, **kwargs)

score_pair(self, x, **kwargs)

class capreolus.reranker.TFCEDRKNRM.TFCEDRKNRM(config=None, provide=None, share_dependency_objects=False, build=True)

Bases: capreolus.reranker.Reranker

TensorFlow implementation of CEDR-KNRM. Equivalant to BERT-KNRM when cls=None.

CEDR: Contextualized Embeddings for Document Ranking Sean MacAvaney, Andrew Yates, Arman Cohan, and Nazli Goharian. SIGIR 2019. https://arxiv.org/pdf/1904.07094

module_name = TFCEDRKNRM

dependencies

config_spec

build_model(self)