SupervisedMarginMSE

class lightning_ir.loss.pairwise.SupervisedMarginMSE

Bases: MarginMSE

Supervised Margin MSE loss for pairwise ranking tasks with a dynamic margin.

__init__()

Initialize the SupervisedMarginMSE loss function.

Methods

__init__()

Initialize the SupervisedMarginMSE loss function.