capreolus.reranker.KNRM

Module Contents

Classes

KNRM_class(extractor, config)
KNRM()
capreolus.reranker.KNRM.logger[source]
class capreolus.reranker.KNRM.KNRM_class(extractor, config)[source]

Bases: torch.nn.Module

get_embedding(self, toks)[source]
forward(self, doctoks, querytoks, query_idf)[source]
class capreolus.reranker.KNRM.KNRM[source]

Bases: capreolus.reranker.Reranker

module_name = KNRM[source]
description = 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.[source]
config_spec[source]
add_summary(self, summary_writer, niter)[source]

Write to the summay_writer custom visualizations/data specific to this reranker

build_model(self)[source]
score(self, d)[source]
test(self, d)[source]