ScoreBasedInBatchLossFunction

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

Bases: InBatchLossFunction

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

Methods

__init__(min_target_diff[, max_num_neg_samples])
compute_loss(output, *args, **kwargs)
get_ib_idcs(output, batch)
process_scores(output)
process_targets(scores, batch)