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.KerasLCEModel(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
- class capreolus.reranker.common.TFLCELoss[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