listwise

Listwise loss functions for the Lightning IR framework.

This module contains loss functions that operate on entire lists of items, considering the ranking of all items simultaneously.

Classes

InfoNCE()

InfoNCE loss for listwise ranking tasks.

KLDivergence()

Kullback-Leibler Divergence loss for listwise ranking tasks.

PearsonCorrelation()

Pearson Correlation loss for listwise ranking tasks.