:mod:`capreolus.reranker.KNRM` ============================== .. py:module:: capreolus.reranker.KNRM Module Contents --------------- Classes ~~~~~~~ .. autoapisummary:: capreolus.reranker.KNRM.KNRM_class capreolus.reranker.KNRM.KNRM .. data:: logger .. py:class:: KNRM_class(extractor, config) Bases: :class:`torch.nn.Module` .. method:: get_embedding(self, toks) .. method:: forward(self, doctoks, querytoks, query_idf) .. py:class:: KNRM Bases: :class:`capreolus.reranker.Reranker` .. attribute:: module_name :annotation: = KNRM .. attribute:: description :annotation: = Chenyan Xiong, Zhuyun Dai, Jamie Callan, Zhiyuan Liu, and Russell Power. 2017. End-to-End Neural Ad-hoc Ranking with Kernel Pooling. In SIGIR'17. .. attribute:: config_spec .. method:: add_summary(self, summary_writer, niter) Write to the summay_writer custom visualizations/data specific to this reranker .. method:: build_model(self) .. method:: score(self, d) .. method:: test(self, d)