capreolus.reranker.ConvKNRM

Module Contents

Classes

ConvKNRM_class

ConvKNRM

Zhuyun Dai, Chenyan Xiong, Jamie Callan, and Zhiyuan Liu. 2018. Convolutional Neural Networks for Soft-Matching N-Grams in Ad-hoc Search. In WSDM'18.

Attributes

logger

capreolus.reranker.ConvKNRM.logger[source]
class capreolus.reranker.ConvKNRM.ConvKNRM_class(extractor, config)[source]

Bases: torch.nn.Module

forward(self, sentence, query_sentence, query_idf)[source]
class capreolus.reranker.ConvKNRM.ConvKNRM(config=None, provide=None, share_dependency_objects=False, build=True)[source]

Bases: capreolus.reranker.Reranker

Zhuyun Dai, Chenyan Xiong, Jamie Callan, and Zhiyuan Liu. 2018. Convolutional Neural Networks for Soft-Matching N-Grams in Ad-hoc Search. In WSDM’18.

module_name = ConvKNRM[source]
dependencies[source]
config_spec[source]
build_model(self)[source]
score(self, d)[source]
test(self, d)[source]
zero_grad(self, *args, **kwargs)[source]