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.
similarity_matrix_tf
(query_embed, doc_embed, query_tok, doc_tok, padding)[source]¶
-
class
capreolus.reranker.common.
RbfKernel
(initial_mu, initial_sigma, requires_grad=True)[source]¶ Bases:
torch.nn.Module
-
class
capreolus.reranker.common.
RbfKernelBank
(mus=None, sigmas=None, dim=1, requires_grad=True)[source]¶ Bases:
torch.nn.Module
-
class
capreolus.reranker.common.
RbfKernelBankTF
(mus, sigmas, dim=1, requires_grad=True, **kwargs)[source]¶ Bases:
tensorflow.keras.layers.Layer