ApproxRankMSE

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

Bases: ApproxLossFunction

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

Methods

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