ConstantMarginMSE

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

Bases: MarginMSE

Constant Margin MSE loss for pairwise ranking tasks with a fixed margin.

__init__(margin: float = 1.0)

Initialize the ConstantMarginMSE loss function.

Parameters:

margin (float) – The fixed margin value for the loss.

Methods

__init__([margin])

Initialize the ConstantMarginMSE loss function.