Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/huggingface/sentence-transformers
/ functions
Functions
3,759 in github.com/huggingface/sentence-transformers
⨍
Functions
3,759
◇
Types & classes
446
↳
Endpoints
48
Method
forward
(self, sentence_features: Iterable[dict[str, Tensor]], labels: Tensor)
sentence_transformers/sentence_transformer/losses/multiple_negatives_ranking.py:232
Method
forward
(self, sentence_features: Iterable[dict[str, Tensor]], labels: Tensor)
sentence_transformers/sentence_transformer/losses/triplet.py:83
Method
forward
(self, sentence_features: Iterable[dict[str, Tensor]], labels: Tensor, size_average=False)
sentence_transformers/sentence_transformer/losses/online_contrastive.py:78
Method
forward
(self, sentence_features: Iterable[dict[str, Tensor]], labels: Tensor)
sentence_transformers/sentence_transformer/losses/gist_embed.py:140
Method
forward
(self, sentence_features: Iterable[dict[str, Tensor]], labels: Tensor)
sentence_transformers/sentence_transformer/losses/cached_multiple_negatives_ranking.py:374
Method
forward
(self, sentence_features: Iterable[dict[str, Tensor]], labels: Tensor)
sentence_transformers/sentence_transformer/losses/batch_hard_soft_margin_triplet.py:93
Method
forward
(self, sentence_features: Iterable[dict[str, Tensor]], labels: Tensor)
sentence_transformers/sentence_transformer/losses/cosent.py:82
Method
forward
(self, sentence_features: Iterable[dict[str, Tensor]], labels: Tensor | None = None)
sentence_transformers/sentence_transformer/losses/contrastive_tension.py:136
Method
forward
(self, sentence_features: Iterable[dict[str, Tensor]], labels: Tensor)
sentence_transformers/sentence_transformer/losses/contrastive_tension.py:261
Method
forward
(self, sentence_features: Iterable[dict[str, Tensor]], labels: Tensor)
sentence_transformers/sentence_transformer/losses/contrastive.py:99
Method
forward
(self, sentence_features: Iterable[dict[str, Tensor]], labels: Tensor)
sentence_transformers/sentence_transformer/losses/margin_mse.py:174
Method
forward
( self, sentence_features: Iterable[dict[str, Tensor]], labels: Tensor | None = None )
sentence_transformers/sentence_transformer/losses/global_orthogonal_regularization.py:134
Method
forward
( self, sentence_features: Iterable[dict[str, Tensor]], labels: Tensor )
sentence_transformers/sentence_transformer/losses/softmax.py:122
Method
forward
(self, sentence_features: Iterable[dict[str, Tensor]], labels: Tensor)
sentence_transformers/sentence_transformer/losses/cosine_similarity.py:79
Method
forward
(self, sentence_features: Iterable[dict[str, Tensor]], labels: Tensor)
sentence_transformers/sentence_transformer/losses/distill_kl_div.py:169
Method
forward
(self, sentence_features: Iterable[dict[str, Tensor]], labels: Tensor)
sentence_transformers/sentence_transformer/losses/batch_all_triplet.py:93
Method
forward
(self, sentence_features: Iterable[dict[str, Tensor]], labels: Tensor)
sentence_transformers/sentence_transformer/losses/batch_hard_triplet.py:150
Method
forward
( self, sentence_features: Iterable[dict[str, Tensor]], labels: Tensor | None = None,
sentence_transformers/multi_vector_encoder/losses/multiple_negatives_ranking.py:140
Method
forward
( self, sentence_features: Iterable[dict[str, Tensor]], labels: Tensor | None = None,
sentence_transformers/multi_vector_encoder/losses/cached_multiple_negatives_ranking.py:290
Method
forward
( self, sentence_features: Iterable[dict[str, Tensor]], labels: Tensor, )
sentence_transformers/multi_vector_encoder/losses/margin_mse.py:119
Method
forward
( self, sentence_features: Iterable[dict[str, Tensor]], labels: Tensor, )
sentence_transformers/multi_vector_encoder/losses/distill_kl_div.py:142
Method
forward
(self, inputs: list[list[str]], labels: Tensor)
sentence_transformers/cross_encoder/fit_mixin.py:152
Method
forward
(self, features: dict[str, torch.Tensor], **kwargs)
sentence_transformers/cross_encoder/modules/logit_score.py:42
Method
forward
Calculate weights for the loss function. Args: gain: Normalized gains tensor discount: Discount tensor
sentence_transformers/cross_encoder/losses/lambda_loss.py:19
Method
forward
(self, gain: Tensor, discount: Tensor, true_sorted: Tensor)
sentence_transformers/cross_encoder/losses/lambda_loss.py:37
Method
forward
(self, gain: Tensor, discount: Tensor, true_sorted: Tensor)
sentence_transformers/cross_encoder/losses/lambda_loss.py:49
Method
forward
(self, gain: Tensor, discount: Tensor, true_sorted: Tensor)
sentence_transformers/cross_encoder/losses/lambda_loss.py:61
Method
forward
(self, gain: Tensor, discount: Tensor, true_sorted: Tensor)
sentence_transformers/cross_encoder/losses/lambda_loss.py:78
Method
forward
(self, gain: Tensor, discount: Tensor, true_sorted: Tensor)
sentence_transformers/cross_encoder/losses/lambda_loss.py:97
Method
forward
Compute LambdaLoss for a batch of queries and their documents. Args: inputs: List of (queries, documents_list)
sentence_transformers/cross_encoder/losses/lambda_loss.py:218
Method
forward
( self, inputs: list[list[str]], labels: Tensor, prompt: str | None = None, task: str | None = None
sentence_transformers/cross_encoder/losses/mse.py:88
Method
forward
( self, inputs: list[list[str]], labels: Tensor, prompt: str | None = None, task: str | None = None
sentence_transformers/cross_encoder/losses/multiple_negatives_ranking.py:173
Method
forward
( self, inputs: list[list[str]], labels: Tensor, prompt: str | None = None, task: str | None = None
sentence_transformers/cross_encoder/losses/cross_entropy.py:68
Method
forward
Calculate position-aware weights for the PListMLE loss. Args: mask: A boolean mask indicating valid positions [batch_siz
sentence_transformers/cross_encoder/losses/plist_mle.py:24
Method
forward
Compute PListMLE loss for a batch of queries and their documents. Args: inputs: List of (queries, documents_list)
sentence_transformers/cross_encoder/losses/plist_mle.py:160
Method
forward
( self, inputs: list[list[str]], labels: Tensor | None = None, prompt: str | None = None, task: str |
sentence_transformers/cross_encoder/losses/cached_multiple_negatives_ranking.py:229
Method
forward
Compute ListNet loss for a batch of queries and their documents. Args: inputs: List of (queries, documents_list)
sentence_transformers/cross_encoder/losses/list_net.py:101
Method
forward
( self, inputs: list[list[str]], labels: Tensor | list[Tensor], prompt: str |
sentence_transformers/cross_encoder/losses/margin_mse.py:103
Method
forward
Compute ADR-MSE loss for a batch of queries and their documents. Args: inputs: List of (queries, documents_list)
sentence_transformers/cross_encoder/losses/adr_mse.py:128
Method
forward
( self, inputs: list[list[str]], labels: Tensor, prompt: str | None = None, task: str | None = None
sentence_transformers/cross_encoder/losses/binary_cross_entropy.py:93
Method
forward
(self, features: dict[str, torch.Tensor])
sentence_transformers/sparse_encoder/modules/sparse_static_embedding.py:65
Method
forward
(self, x: torch.Tensor)
sentence_transformers/sparse_encoder/modules/sparse_auto_encoder.py:20
Method
forward
( self, features: dict[str, torch.Tensor], max_active_dims: int | None = None )
sentence_transformers/sparse_encoder/modules/sparse_auto_encoder.py:161
Method
forward
Forward pass of the model. Args: features: Dictionary containing input features. Expects: - 'token_embedd
sentence_transformers/sparse_encoder/modules/splade_pooling.py:66
Method
forward
(self, sentence_features: Iterable[dict[str, Tensor]], labels: Tensor)
sentence_transformers/sparse_encoder/losses/flops.py:42
Method
forward
(self, sentence_features: Iterable[dict[str, Tensor]], labels: Tensor)
sentence_transformers/sparse_encoder/losses/sparse_multiple_negatives_ranking.py:141
Method
forward
( self, sentence_features: Iterable[dict[str, Tensor]], labels: Tensor | None = None )
sentence_transformers/sparse_encoder/losses/cached_splade.py:197
Method
forward
(self, sentence_features: Iterable[dict[str, Tensor]], labels: Tensor)
sentence_transformers/sparse_encoder/losses/sparse_angle.py:85
Method
forward
(self, sentence_features: Iterable[dict[str, Tensor]], labels: Tensor)
sentence_transformers/sparse_encoder/losses/sparse_margin_mse.py:176
Method
forward
(self, sentence_features: Iterable[dict[str, Tensor]], labels: Tensor)
sentence_transformers/sparse_encoder/losses/sparse_triplet.py:70
Method
forward
(self, sentence_features: Iterable[dict[str, Tensor]], labels: Tensor)
sentence_transformers/sparse_encoder/losses/sparse_cosine_similarity.py:76
Method
forward
( self, sentence_features: Iterable[dict[str, torch.Tensor]], labels: torch.Tensor | None = None )
sentence_transformers/sparse_encoder/losses/splade.py:138
Method
forward
(self, sentence_features: Iterable[dict[str, Tensor]], labels: Tensor)
sentence_transformers/sparse_encoder/losses/sparse_distill_kl_div.py:157
Method
forward
(self, sentence_features: Iterable[dict[str, Tensor]], labels: Tensor)
sentence_transformers/sparse_encoder/losses/sparse_cosent.py:75
Method
forward
(self, sentence_features: Iterable[dict[str, torch.Tensor]])
sentence_transformers/sparse_encoder/losses/csr.py:63
Method
forward
( self, sentence_features: Iterable[dict[str, torch.Tensor]], labels: torch.Tensor | None = None )
sentence_transformers/sparse_encoder/losses/csr.py:189
Method
forward
Forward pass through all modules in the model.
sentence_transformers/base/model.py:554
Method
forward
(self, features: dict[str, Tensor])
sentence_transformers/base/modules/normalize.py:39
Method
forward
Forward pass of the module. This method should be overridden by subclasses to implement the specific behavior of the module. The for
sentence_transformers/base/modules/module.py:87
Method
forward
(self, features, *args, **kwargs)
tests/sentence_transformer/test_trainer.py:389
Method
forward
(self, *args, **kwargs)
tests/sentence_transformer/losses/test_adaptive_layer.py:23
Method
forward
(self, *args, **kwargs)
tests/sentence_transformer/losses/test_adaptive_layer.py:35
Method
forward
(self, sentence_features, labels)
tests/sentence_transformer/losses/test_adaptive_layer.py:214
Method
forward
(self, sentence_features, labels)
tests/sentence_transformer/losses/test_gradcache.py:234
Method
forward
(self, *args, **kwargs)
tests/sentence_transformer/losses/test_matryoshka.py:16
Method
forward
(self, features)
tests/base/modules/test_router.py:50
Method
forward
(self, features)
tests/base/modules/test_router.py:90
Method
forward
(self, features, **kwargs)
tests/base/modules/test_router.py:570
Method
forward
(self, features)
tests/base/modules/test_input_module.py:13
Method
forward
(self, features)
tests/base/modules/test_input_module.py:26
Method
forward
(self, features)
tests/base/modules/test_input_module.py:41
Method
forward
(self, sentence_features, labels)
examples/sentence_transformer/training/ms_marco/train_bi_encoder_margin_mse_mnrl.py:182
Method
forward
( self, sentence_features: Iterable[dict[str, Tensor]], labels: Tensor | None = None,
examples/sentence_transformer/training/other/training_gooaq_infonce_gor.py:58
Method
forward_mini_batched
(self, sentence_features: Iterable[dict[str, Tensor]], labels: Tensor)
sentence_transformers/sentence_transformer/losses/mega_batch_margin.py:191
Method
forward_non_mini_batched
(self, sentence_features: Iterable[dict[str, Tensor]], labels: Tensor)
sentence_transformers/sentence_transformer/losses/mega_batch_margin.py:197
Function
forward_tracker
(inputs, labels, prompt=None, task=None)
tests/cross_encoder/test_trainer.py:267
Function
forward_without_trailing_dim
(features, **kwargs)
tests/cross_encoder/test_model.py:259
Method
from_input_examples
(cls, examples: list[InputExample], **kwargs)
sentence_transformers/sentence_transformer/evaluation/binary_classification.py:140
Method
from_input_examples
(cls, examples: list[InputExample], **kwargs)
sentence_transformers/sentence_transformer/evaluation/triplet.py:164
Method
from_input_examples
(cls, examples: list[InputExample], **kwargs)
sentence_transformers/sentence_transformer/evaluation/embedding_similarity.py:142
Method
from_input_examples
(cls, examples: list[InputExample], **kwargs)
sentence_transformers/cross_encoder/evaluation/deprecated.py:22
Method
from_input_examples
(cls, examples: list[InputExample], **kwargs)
sentence_transformers/cross_encoder/evaluation/deprecated.py:43
Method
from_input_examples
(cls, examples: list[InputExample], **kwargs)
sentence_transformers/cross_encoder/evaluation/deprecated.py:64
Method
from_input_examples
(cls, examples: list[InputExample], **kwargs)
sentence_transformers/cross_encoder/evaluation/deprecated.py:85
Method
from_input_examples
(cls, examples: list[InputExample], **kwargs)
sentence_transformers/cross_encoder/evaluation/correlation.py:91
Method
func
(**kwargs)
tests/util/test_decorators.py:27
Method
func
(**kwargs)
tests/util/test_decorators.py:100
Method
func
(self, **kwargs)
tests/util/test_decorators.py:201
Method
func
(self, *args, **kwargs)
tests/util/test_decorators.py:371
Method
gather_across_devices
(self)
sentence_transformers/sentence_transformer/losses/cached_multiple_negatives_ranking.py:225
Method
generate_usage_snippet
(self)
sentence_transformers/sentence_transformer/model_card.py:147
Method
generate_usage_snippet
(self)
sentence_transformers/multi_vector_encoder/model_card.py:142
Method
get_adapter_state_dict
If you are not familiar with adapters and PEFT methods, we invite you to read more about them on the PEFT official documentation: htt
sentence_transformers/base/peft_mixin.py:131
Method
get_config_dict
(self)
sentence_transformers/sentence_transformer/losses/cached_gist_embed.py:421
Method
get_config_dict
(self)
sentence_transformers/sentence_transformer/losses/matryoshka_2d.py:142
Method
get_config_dict
(self)
sentence_transformers/sentence_transformer/losses/denoising_auto_encoder.py:217
Method
get_config_dict
(self)
sentence_transformers/sentence_transformer/losses/matryoshka.py:264
Method
get_config_dict
(self)
sentence_transformers/sentence_transformer/losses/batch_semi_hard_triplet.py:177
Method
get_config_dict
(self)
sentence_transformers/sentence_transformer/losses/multiple_negatives_ranking.py:340
Method
get_config_dict
(self)
sentence_transformers/sentence_transformer/losses/triplet.py:105
← previous
next →
1,501–1,600 of 3,759, ranked by callers