base

Base classes and abstract interfaces for loss functions in the Lightning IR framework.

This module defines the abstract base classes and common functionality for all loss functions used in the Lightning IR framework.

Classes

EmbeddingLossFunction()

Base class for loss functions that operate on embeddings.

InBatchLossFunction([...])

Base class for in-batch loss functions that compute in-batch indices for positive and negative samples.

ListwiseLossFunction()

Base class for listwise loss functions.

LossFunction()

Base class for loss functions in the Lightning IR framework.

PairwiseLossFunction()

Base class for pairwise loss functions.

RegularizationLossFunction([query_weight, ...])

Base class for regularization loss functions that operate on embeddings.

ScoringLossFunction()

Base class for loss functions that operate on scores.