ApproxRankMSE

class lightning_ir.loss.loss.ApproxRankMSE(temperature: float = 1, discount: Literal['log2', 'reciprocal'] | None = None)

Bases: ApproxLossFunction

__init__(temperature: float = 1, discount: Literal['log2', 'reciprocal'] | None = None)

Methods

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