ApproxMRR

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

Bases: ApproxLossFunction

__init__(temperature: float = 1)[source]

Methods

__init__([temperature])

compute_loss(output, batch)

get_mrr(ranks, targets[, k])