capreolus.reranker.TFKNRM

Module Contents

Classes

TFKNRM_Class
TFKNRM TensorFlow implementation of KNRM.
class capreolus.reranker.TFKNRM.TFKNRM_Class(extractor, config, **kwargs)[source]

Bases: tensorflow.keras.layers.Layer

get_score(self, doc_tok, query_tok, query_idf)[source]
call(self, x, **kwargs)[source]
predict_step(self, data)[source]
score(self, x, **kwargs)[source]
score_pair(self, x, **kwargs)[source]
class capreolus.reranker.TFKNRM.TFKNRM(config=None, provide=None, share_dependency_objects=False, build=True)[source]

Bases: capreolus.reranker.Reranker

TensorFlow implementation of KNRM.

Chenyan Xiong, Zhuyun Dai, Jamie Callan, Zhiyuan Liu, and Russell Power. 2017. End-to-End Neural Ad-hoc Ranking with Kernel Pooling. In SIGIR‘17.

module_name = TFKNRM[source]
dependencies[source]
config_spec[source]
build_model(self)[source]