View source on GitHub
|
Keras losses in TF-Ranking.
Classes
class ApproxMRRLoss: Computes approximate MRR loss between y_true and y_pred.
class ApproxNDCGLoss: Computes approximate NDCG loss between y_true and y_pred.
class ClickEMLoss: Computes click EM loss between y_true and y_pred.
class CoupledRankDistilLoss: Computes the Rank Distil loss between y_true and y_pred.
class DCGLambdaWeight: Keras serializable class for DCG.
class GumbelApproxNDCGLoss: Computes the Gumbel approximate NDCG loss between y_true and y_pred.
class LabelDiffLambdaWeight: Keras serializable class for LabelDiffLambdaWeight.
class ListMLELambdaWeight: LambdaWeight for ListMLE cost function.
class ListMLELoss: Computes ListMLE loss between y_true and y_pred.