ScoreBasedInBatchLossFunction

class lightning_ir.loss.in_batch.ScoreBasedInBatchLossFunction(min_target_diff: float, max_num_neg_samples: int | None = None)[source]

Bases: InBatchLossFunction

Base class for in-batch loss functions that compute in-batch indices based on scores.

__init__(min_target_diff: float, max_num_neg_samples: int | None = None)[source]

Initialize the ScoreBasedInBatchLossFunction.

Parameters:

  • min_target_diff (float) – Minimum target difference for negative sampling.

  • max_num_neg_samples (int | None) – Maximum number of negative samples.

Methods

__init__(min_target_diff[, max_num_neg_samples])

Initialize the ScoreBasedInBatchLossFunction.