capreolus.reranker.HINT

Module Contents

Classes

GRUCell2d

GRUModel2d

HiNT

HiNT_main

HINT

Yixing Fan, Jiafeng Guo, Yanyan Lan, Jun Xu, Chengxiang Zhai, and Xueqi Cheng. 2018. Modeling Diverse Relevance Patterns in Ad-hoc Retrieval. In SIGIR'18.

Attributes

device

capreolus.reranker.HINT.device[source]
class capreolus.reranker.HINT.GRUCell2d(input_size, hidden_size, bias=True)[source]

Bases: torch.nn.Module

reset_parameters()[source]
softmax_by_row(zi, zl, zt, zd)[source]
forward(x, hidden_i1_j1, hidden_i1_j, hidden_i_j1)[source]
class capreolus.reranker.HINT.GRUModel2d(input_dim, hidden_dim, bias=True)[source]

Bases: torch.nn.Module

forward(x)[source]
class capreolus.reranker.HINT.HiNT(weights_matrix, p)[source]

Bases: torch.nn.Module

init_hidden()[source]
matrix_inv(A)[source]
forward(sentence, query_sentence, M_XOR, M_cos, masks)[source]

M_XOR or M_cos: (B, Q, D) masks: (B, Q, D) have 0 on non-pad positions and 1 on pad positions

class capreolus.reranker.HINT.HiNT_main(extractor, config)[source]

Bases: torch.nn.Module

init_hidden()[source]
forward(query_sentence, query_idf, pos_sentence, neg_sentence)[source]
test_forward(query_sentence, query_idf, pos_sentence)[source]
class capreolus.reranker.HINT.HINT(config=None, provide=None, share_dependency_objects=False, build=True)[source]

Bases: capreolus.reranker.Reranker

Yixing Fan, Jiafeng Guo, Yanyan Lan, Jun Xu, Chengxiang Zhai, and Xueqi Cheng. 2018. Modeling Diverse Relevance Patterns in Ad-hoc Retrieval. In SIGIR’18.

module_name = 'HINT'[source]
dependencies[source]
config_spec[source]
test(query_sentence, query_idf, pos_sentence, *args, **kwargs)[source]
score(d)[source]
test(d)[source]
zero_grad(*args, **kwargs)[source]
build_model()[source]