:mod:`capreolus.reranker.common` ================================ .. py:module:: capreolus.reranker.common Module Contents --------------- Classes ~~~~~~~ .. autoapisummary:: capreolus.reranker.common.SimilarityMatrix capreolus.reranker.common.RbfKernel capreolus.reranker.common.RbfKernelBank capreolus.reranker.common.RbfKernelBankTF capreolus.reranker.common.RbfKernelTF Functions ~~~~~~~~~ .. autoapisummary:: capreolus.reranker.common.pair_softmax_loss capreolus.reranker.common.pair_hinge_loss capreolus.reranker.common.similarity_matrix_tf capreolus.reranker.common.create_emb_layer .. function:: pair_softmax_loss(pos_neg_scores) .. function:: pair_hinge_loss(pos_neg_scores) .. function:: similarity_matrix_tf(query_embed, doc_embed, query_tok, doc_tok, padding) .. py:class:: SimilarityMatrix(padding=0) Bases: :class:`torch.nn.Module` .. method:: forward(self, query_embed, doc_embed, query_tok, doc_tok) .. py:class:: RbfKernel(initial_mu, initial_sigma, requires_grad=True) Bases: :class:`torch.nn.Module` .. method:: forward(self, data) .. py:class:: RbfKernelBank(mus=None, sigmas=None, dim=1, requires_grad=True) Bases: :class:`torch.nn.Module` .. method:: count(self) .. method:: forward(self, data) .. py:class:: RbfKernelBankTF(mus, sigmas, dim=1, requires_grad=True, **kwargs) Bases: :class:`tensorflow.keras.layers.Layer` .. method:: count(self) .. method:: call(self, data, **kwargs) .. py:class:: RbfKernelTF(initial_mu, initial_sigma, requires_grad=True, **kwargs) Bases: :class:`tensorflow.keras.layers.Layer` .. method:: call(self, data, *kwargs) .. function:: create_emb_layer(weights, non_trainable=True)