lsmpp's picture
Add files using upload-large-folder tool
bd33eac verified

Losses

sentence_transformers.cross_encoder.losses defines different loss functions that can be used to fine-tune cross-encoder models on training data. The choice of loss function plays a critical role when fine-tuning the model. It determines how well our model will work for the specific downstream task.

Sadly, there is no "one size fits all" loss function. Which loss function is suitable depends on the available training data and on the target task. Consider checking out the Loss Overview to help narrow down your choice of loss function(s).

BinaryCrossEntropyLoss

.. autoclass:: sentence_transformers.cross_encoder.losses.BinaryCrossEntropyLoss

CrossEntropyLoss

.. autoclass:: sentence_transformers.cross_encoder.losses.CrossEntropyLoss

LambdaLoss

.. autoclass:: sentence_transformers.cross_encoder.losses.LambdaLoss

.. autoclass:: sentence_transformers.cross_encoder.losses.LambdaLoss.BaseWeightingScheme
.. autoclass:: sentence_transformers.cross_encoder.losses.NoWeightingScheme
.. autoclass:: sentence_transformers.cross_encoder.losses.NDCGLoss1Scheme
.. autoclass:: sentence_transformers.cross_encoder.losses.NDCGLoss2Scheme
.. autoclass:: sentence_transformers.cross_encoder.losses.LambdaRankScheme
.. autoclass:: sentence_transformers.cross_encoder.losses.NDCGLoss2PPScheme

ListMLELoss

.. autoclass:: sentence_transformers.cross_encoder.losses.ListMLELoss

PListMLELoss

.. autoclass:: sentence_transformers.cross_encoder.losses.PListMLELoss

.. autoclass:: sentence_transformers.cross_encoder.losses.PListMLELambdaWeight

ListNetLoss

.. autoclass:: sentence_transformers.cross_encoder.losses.ListNetLoss

MultipleNegativesRankingLoss

.. autoclass:: sentence_transformers.cross_encoder.losses.MultipleNegativesRankingLoss

CachedMultipleNegativesRankingLoss

.. autoclass:: sentence_transformers.cross_encoder.losses.CachedMultipleNegativesRankingLoss

MSELoss

.. autoclass:: sentence_transformers.cross_encoder.losses.MSELoss

MarginMSELoss

.. autoclass:: sentence_transformers.cross_encoder.losses.MarginMSELoss

RankNetLoss

.. autoclass:: sentence_transformers.cross_encoder.losses.RankNetLoss