capreolus.reranker.KNRM

Module Contents

Classes

KNRM_class

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.

Attributes

logger

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(config=None, provide=None, share_dependency_objects=False, build=True)[source]

Bases: capreolus.reranker.Reranker

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 = KNRM[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]