tfr.keras.losses.NDCGLambdaWeight

View source on GitHub

Keras serializable class for NDCG.

Inherits From: DCGLambdaWeight

tfr.keras.losses.NDCGLambdaWeight(
    topn=None,
    gain_fn=None,
    rank_discount_fn=None,
    smooth_fraction=0.0,
    **kwargs
)

Args
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

get_config()

individual_weights

View source

individual_weights(
    labels, ranks
)

See _LambdaWeight.

pair_weights

View source

pair_weights(
    labels, ranks
)

See _LambdaWeight.