InBatchLossFunction

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

Bases: LossFunction

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

Methods

__init__([pos_sampling_technique, ...])
get_ib_idcs(output, batch)