capreolus.reranker.common

Module Contents

Classes

SimilarityMatrix(padding=0)
RbfKernel(initial_mu, initial_sigma, requires_grad=True)
RbfKernelBank(mus=None, sigmas=None, dim=1, requires_grad=True)
RbfKernelBankTF(mus, sigmas, dim=1, requires_grad=True, **kwargs)
RbfKernelTF(initial_mu, initial_sigma, requires_grad=True, **kwargs)

Functions

pair_softmax_loss(pos_neg_scores)
pair_hinge_loss(pos_neg_scores)
similarity_matrix_tf(query_embed, doc_embed, query_tok, doc_tok, padding)
create_emb_layer(weights, non_trainable=True)
capreolus.reranker.common.pair_softmax_loss(pos_neg_scores)[source]
capreolus.reranker.common.pair_hinge_loss(pos_neg_scores)[source]
capreolus.reranker.common.similarity_matrix_tf(query_embed, doc_embed, query_tok, doc_tok, padding)[source]
class capreolus.reranker.common.SimilarityMatrix(padding=0)[source]

Bases: torch.nn.Module

forward(self, query_embed, doc_embed, query_tok, doc_tok)[source]
class capreolus.reranker.common.RbfKernel(initial_mu, initial_sigma, requires_grad=True)[source]

Bases: torch.nn.Module

forward(self, data)[source]
class capreolus.reranker.common.RbfKernelBank(mus=None, sigmas=None, dim=1, requires_grad=True)[source]

Bases: torch.nn.Module

count(self)[source]
forward(self, data)[source]
class capreolus.reranker.common.RbfKernelBankTF(mus, sigmas, dim=1, requires_grad=True, **kwargs)[source]

Bases: tensorflow.keras.layers.Layer

count(self)[source]
call(self, data, **kwargs)[source]
class capreolus.reranker.common.RbfKernelTF(initial_mu, initial_sigma, requires_grad=True, **kwargs)[source]

Bases: tensorflow.keras.layers.Layer

call(self, data, *kwargs)[source]
capreolus.reranker.common.create_emb_layer(weights, non_trainable=True)[source]