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(output, batch)
get_approx_ranks(scores, temperature)
get_dcg(ranks, targets[, k, scale_gains])
get_ndcg(ranks, targets[, k, scale_gains, ...])
process_scores(output)
process_targets(scores, batch)