ConstantMarginMSE

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

Bases: MarginMSE

__init__(margin: float = 1.0)

Methods

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