capreolus.reranker.DUET

Module Contents

Classes

LocalModel
DistributedModel
DUET_class
DUET Bhaskar Mitra, Fernando Diaz, and Nick Craswell. 2017. Learning to Match using Local and Distributed Representations of Text for Web Search. In WWW‘17.
capreolus.reranker.DUET.logger[source]
class capreolus.reranker.DUET.LocalModel(p)[source]

Bases: torch.nn.Module

exact_match(self, m1, m2)[source]

m1: (B, len1) m2: (B, len2)

forward(self, documents, queries, query_idf)[source]

queries: (B, nq) documents: (B, nd) query_idf: (B, nq)

class capreolus.reranker.DUET.DistributedModel(extractor, p)[source]

Bases: torch.nn.Module

forward(self, documents, queries)[source]
class capreolus.reranker.DUET.DUET_class(extractor, p)[source]

Bases: torch.nn.Module

forward(self, documents, queries, query_idf)[source]

queries: (B, nq) documents: (B, nd)

class capreolus.reranker.DUET.DUET(config=None, provide=None, share_dependency_objects=False, build=True)[source]

Bases: capreolus.reranker.Reranker

Bhaskar Mitra, Fernando Diaz, and Nick Craswell. 2017. Learning to Match using Local and Distributed Representations of Text for Web Search. In WWW‘17.

module_name = DUET[source]
dependencies[source]
config_spec[source]
build_model(self)[source]
score(self, d)[source]
test(self, d)[source]