ApproxMRR

class lightning_ir.loss.loss.ApproxMRR(temperature: float = 1)[source]

Bases: ApproxLossFunction

__init__(temperature: float = 1)[source]

Methods

__init__([temperature])

compute_loss(scores, targets)

get_approx_ranks(scores, temperature)

get_mrr(ranks, targets[, k])

process_targets(scores, targets)