InBatchLossFunction

class lightning_ir.loss.loss.InBatchLossFunction(pos_sampling_technique: Literal['all', 'first'] = 'all', neg_sampling_technique: Literal['all', 'first'] = 'all', max_num_neg_samples: int | None = None)

Bases: ScoringLossFunction

__init__(pos_sampling_technique: Literal['all', 'first'] = 'all', neg_sampling_technique: Literal['all', 'first'] = 'all', max_num_neg_samples: int | None = None)

Methods

__init__([pos_sampling_technique, ...])
compute_loss(scores)
get_ib_idcs(num_queries, num_docs)
process_targets(scores, targets)