ApproxLossFunction

class lightning_ir.loss.loss.ApproxLossFunction(temperature: float = 1)

Bases: ListwiseLossFunction

__init__(temperature: float = 1) → None

Methods

__init__([temperature])
compute_loss(scores, targets)
get_approx_ranks(scores, temperature)
process_targets(scores, targets)