capreolus.reranker.DRMM

Module Contents

Classes

DRMM_class
DRMM	Jiafeng Guo, Yixing Fan, Qingyao Ai, and W. Bruce Croft. 2016. A Deep Relevance Matching Model for Ad-hoc Retrieval. In CIKM'16.

Attributes

logger
dtype

capreolus.reranker.DRMM.logger

class capreolus.reranker.DRMM.DRMM_class(extractor, config)

Bases: torch.nn.Module

forward(sentence, query_sentence, query_idf)

capreolus.reranker.DRMM.dtype

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

Bases: capreolus.reranker.Reranker

Jiafeng Guo, Yixing Fan, Qingyao Ai, and W. Bruce Croft. 2016. A Deep Relevance Matching Model for Ad-hoc Retrieval. In CIKM’16.

module_name = 'DRMM'

config_spec

build_model()

score(d)

test(d)