ApproxNDCG

class lightning_ir.loss.loss.ApproxNDCG(temperature: float = 1, scale_gains: bool = True)

Bases: ApproxLossFunction

__init__(temperature: float = 1, scale_gains: bool = True)

Methods

__init__([temperature, scale_gains])
compute_loss(scores, targets)
get_approx_ranks(scores, temperature)
get_dcg(ranks, targets[, k, scale_gains])
get_ndcg(ranks, targets[, k, scale_gains, ...])
process_targets(scores, targets)