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