RegularizationLossFunction

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

Bases: EmbeddingLossFunction

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

Methods

__init__([query_weight, doc_weight])
compute_loss(query_embeddings, doc_embeddings)