ApproxLossFunction

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

Bases: ListwiseLossFunction

__init__(temperature: float = 1) → None

Methods

__init__([temperature])
compute_loss(output, batch)
get_approx_ranks(scores, temperature)
process_scores(output)
process_targets(scores, batch)