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

class capreolus.reranker.KNRM.KNRM_class(extractor, config)

Bases: torch.nn.Module

get_embedding(toks)

forward(doctoks, querytoks, query_idf)

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

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'

config_spec

add_summary(summary_writer, niter)

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

build_model()

score(d)

test(d)