capreolus.reranker.common
¶
Module Contents¶
Classes¶
Functions¶
|
|
|
|
|
|
|
Original TF similarity matrix. May have issues with mixed precision. Use new_similarity_matrix_tf instead |
|
-
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.
new_similarity_matrix_tf
(query_embed, doc_embed, query_tok, doc_tok, padding)[source]¶
-
capreolus.reranker.common.
similarity_matrix_tf
(query_embed, doc_embed, query_tok, doc_tok, padding)[source]¶ Original TF similarity matrix. May have issues with mixed precision. Use new_similarity_matrix_tf instead
-
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