ApproxMRR

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

Methods

`__init__`([temperature])
`compute_loss`(output, batch)
`get_approx_ranks`(scores, temperature)
`get_mrr`(ranks, targets[, k])
`process_scores`(output)
`process_targets`(scores, batch)