capreolus.reranker.common
¶
Module Contents¶
Classes¶
KerasPairModel |
|
KerasTripletModel |
|
TFPairwiseHingeLoss |
|
TFCategoricalCrossEntropyLoss |
|
SimilarityMatrix |
|
StackedSimilarityMatrix |
|
RbfKernel |
|
RbfKernelBank |
|
RbfKernelBankTF |
|
RbfKernelTF |
Functions¶
pair_softmax_loss (pos_neg_scores, *args, **kwargs) |
|
pair_hinge_loss (pos_neg_scores, *args, **kwargs) |
|
similarity_matrix_tf (query_embed, doc_embed, query_tok, doc_tok, padding) |
|
create_emb_layer (weights, non_trainable=True) |
-
class
capreolus.reranker.common.
KerasPairModel
(model, *args, **kwargs)[source]¶ Bases:
tensorflow.keras.Model
-
class
capreolus.reranker.common.
KerasTripletModel
(model, *args, **kwargs)[source]¶ Bases:
tensorflow.keras.Model
-
class
capreolus.reranker.common.
TFPairwiseHingeLoss
[source]¶ Bases:
tensorflow_ranking.python.keras.losses.PairwiseHingeLoss
-
class
capreolus.reranker.common.
TFCategoricalCrossEntropyLoss
[source]¶ Bases:
tensorflow.python.keras.losses.CategoricalCrossentropy
-
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