pairwise
Pairwise loss functions for the Lightning IR framework.
This module contains loss functions that operate on pairs of items, comparing positive and negative examples.
Classes
|
Constant Margin MSE loss for pairwise ranking tasks with a fixed margin. |
|
Mean Squared Error loss with a margin for pairwise ranking tasks. |
|
RankNet loss function for pairwise ranking tasks. |
Supervised Margin MSE loss for pairwise ranking tasks with a dynamic margin. |