:mod:`capreolus.reranker.HINT` ============================== .. py:module:: capreolus.reranker.HINT Module Contents --------------- Classes ~~~~~~~ .. autoapisummary:: capreolus.reranker.HINT.GRUCell2d capreolus.reranker.HINT.GRUModel2d capreolus.reranker.HINT.HiNT capreolus.reranker.HINT.HiNT_main capreolus.reranker.HINT.HINT .. data:: device .. py:class:: GRUCell2d(input_size, hidden_size, bias=True) Bases: :class:`torch.nn.Module` .. method:: reset_parameters(self) .. method:: softmax_by_row(self, zi, zl, zt, zd) .. method:: forward(self, x, hidden_i1_j1, hidden_i1_j, hidden_i_j1) .. py:class:: GRUModel2d(input_dim, hidden_dim, bias=True) Bases: :class:`torch.nn.Module` .. method:: forward(self, x) .. py:class:: HiNT(weights_matrix, p) Bases: :class:`torch.nn.Module` .. method:: init_hidden(self) .. method:: matrix_inv(self, A) .. method:: forward(self, sentence, query_sentence, M_XOR, M_cos, masks) M_XOR or M_cos: (B, Q, D) masks: (B, Q, D) have 0 on non-pad positions and 1 on pad positions .. py:class:: HiNT_main(extractor, config) Bases: :class:`torch.nn.Module` .. method:: init_hidden(self) .. method:: forward(self, query_sentence, query_idf, pos_sentence, neg_sentence) .. method:: test_forward(self, query_sentence, query_idf, pos_sentence) .. py:class:: HINT Bases: :class:`capreolus.reranker.Reranker` .. attribute:: module_name :annotation: = HINT .. attribute:: description :annotation: = 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. .. attribute:: config_spec .. method:: test(self, query_sentence, query_idf, pos_sentence, *args, **kwargs) .. method:: score(self, d) .. method:: zero_grad(self, *args, **kwargs) .. method:: build_model(self)