InBatchLossFunction

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

Bases: LossFunction

__init__(pos_sampling_technique: 'all' | 'first' = 'all', neg_sampling_technique: 'all' | 'first' | 'all_and_non_first' = 'all', max_num_neg_samples: int | None = None)

Methods

__init__([pos_sampling_technique, ...])
compute_loss(output, *args, **kwargs)
get_ib_idcs(output, batch)
process_scores(output)
process_targets(scores, batch)