ScoreBasedInBatchCrossEntropy

class lightning_ir.loss.loss.ScoreBasedInBatchCrossEntropy(min_target_diff: float, max_num_neg_samples: int | None = None)

Bases: ScoreBasedInBatchLossFunction

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

Methods

compute_loss(output)