tf.keras.losses.get
Retrieves a Keras loss as a function
/Loss
class instance.
tf.keras.losses.get(
identifier
)
The identifier
may be the string name of a loss function or Loss
class.
loss = tf.keras.losses.get("categorical_crossentropy")
type(loss)
<class 'function'>
loss = tf.keras.losses.get("CategoricalCrossentropy")
type(loss)
<class '...keras.losses.CategoricalCrossentropy'>
You can also specify config
of the loss to this function by passing dict
containing class_name
and config
as an identifier. Also note that the
class_name
must map to a Loss
class
identifier = {"class_name": "CategoricalCrossentropy",
"config": {"from_logits": True} }
loss = tf.keras.losses.get(identifier)
type(loss)
<class '...keras.losses.CategoricalCrossentropy'>
Args |
identifier
|
A loss identifier. One of None or string name of a loss
function/class or loss configuration dictionary or a loss function or a
loss class instance.
|
Returns |
A Keras loss as a function / Loss class instance.
|
Raises |
ValueError
|
If identifier cannot be interpreted.
|
Except as otherwise noted, the content of this page is licensed under the Creative Commons Attribution 4.0 License, and code samples are licensed under the Apache 2.0 License. For details, see the Google Developers Site Policies. Java is a registered trademark of Oracle and/or its affiliates. Some content is licensed under the numpy license.
Last updated 2023-10-06 UTC.
[null,null,["Last updated 2023-10-06 UTC."],[],[]]