InBatchCrossEntropy
- class lightning_ir.loss.loss.InBatchCrossEntropy(pos_sampling_technique: Literal['all', 'first'] = 'all', neg_sampling_technique: Literal['all', 'first'] = 'all', max_num_neg_samples: int | None = None)[source]
Bases:
InBatchLossFunction
- __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)