in_batch
In-batch loss functions for the Lightning IR framework.
This module contains loss functions that operate on batches of data, comparing examples within the same batch for training.
Classes
|
In-batch cross-entropy loss function for ranking tasks. |
|
In-batch cross-entropy loss function based on scores for ranking tasks. |
|
Base class for in-batch loss functions that compute in-batch indices based on scores. |