SupervisedMarginMSE

class lightning_ir.loss.loss.SupervisedMarginMSE[source]

Bases: MarginMSE

__init__()[source]

Methods

__init__()

compute_loss(output, batch)

get_pairwise_idcs(targets)

process_scores(output)

process_targets(scores, batch)