RegularizationLossFunction

class lightning_ir.loss.loss.RegularizationLossFunction(query_weight: float = 0.0001, doc_weight: float = 0.0001)[source]

__init__(query_weight: float = 0.0001, doc_weight: float = 0.0001) → None[source]

Methods

`__init__`([query_weight, doc_weight])
`compute_loss`(output)
`process_embeddings`(output)
`process_scores`(output)
`process_targets`(scores, batch)