MarginMSE

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

Bases: PairwiseLossFunction

__init__(margin: float | 'scores' = 1.0)

Methods

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