InBatchCrossEntropy

class lightning_ir.loss.loss.InBatchCrossEntropy(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: InBatchLossFunction

__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

compute_loss(output)