MarginMSE

class lightning_ir.loss.loss.MarginMSE(margin: float | Literal['scores'] = 1.0)

Bases: PairwiseLossFunction

__init__(margin: float | Literal['scores'] = 1.0)

Methods

__init__([margin])
compute_loss(scores, targets)
get_pairwise_idcs(targets)
process_targets(scores, targets)