Factory method to get a ranking loss class.
tfr.keras.losses.get(
loss: str,
reduction: tf.losses.Reduction = tf.losses.Reduction.AUTO,
lambda_weight: Optional[losses_impl._LambdaWeight] = None,
name: Optional[str] = None,
**kwargs
) -> tf.keras.losses.Loss
Args |
loss
|
(str) An attribute of RankingLossKey , defining which loss object to
return.
|
reduction
|
(enum) An enum of strings indicating the loss reduction type.
See type definition in the tf.compat.v2.losses.Reduction .
|
lambda_weight
|
(losses_impl._LambdaWeight) A lambda object for ranking
metric optimization.
|
name
|
(optional) (str) Name of loss.
|
**kwargs
|
Keyword arguments for the loss object.
|
Returns |
A ranking loss instance. See _RankingLoss signature for more details.
|
Raises |
ValueError
|
If loss_key is unsupported.
|