View source on GitHub |
LambdaWeight for ListMLE cost function.
tfr.keras.losses.ListMLELambdaWeight(
rank_discount_fn: Optional[tfr.keras.utils.GainFunction
] = None,
**kwargs
)
Args | |
---|---|
rank_discount_fn
|
(function) The rank discount function. |
Methods
get_config
get_config() -> Dict[str, Any]
individual_weights
individual_weights(
labels, ranks
)
See _LambdaWeight
.
pair_weights
pair_weights(
labels, ranks
)
See _LambdaWeight
.