tfr.keras.losses.NDCGLambdaWeight

Keras serializable class for NDCG.

Inherits From: DCGLambdaWeight

topn (int) The topn for the DCG metric.
gain_fn (function) Transforms labels.
rank_discount_fn (function) The rank discount function.
normalized (bool) If True, normalize weight by the max DCG.
smooth_fraction (float) parameter to control the contribution from LambdaMART.

Methods

get_config

View source

individual_weights

View source

See _LambdaWeight.

pair_weights

View source

See _LambdaWeight.