ConstantMarginMSE

class lightning_ir.loss.loss.ConstantMarginMSE(margin: float = 1.0)

Bases: MarginMSE

__init__(margin: float = 1.0)

Methods

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